2023-10-27 19:19:22 -06:00
1 changed files with 4 additions and 0 deletions
--- a/server.py
+++ b/server.py
@ -99,6 +99,10 @@ def home():
    base_client_api = redis.get('base_client_api', dtype=str)
    stats = generate_stats()
    model_choices, default_model = get_model_choices()
+
+    if not model_choices.get(default_model):
+        return 'The server is still starting up. Please wait...'
+
    default_model_info = model_choices[default_model]

    if default_model_info['queued'] == 0 and default_model_info['queued'] >= default_model_info['concurrent_gens']: