Update app.py

app.py

@@ -37,13 +37,14 @@ global_data = {
 class ModelManager:
     def __init__(self):
         self.models = {}
-        self.load_all_models()
-
-    def load_all_models(self):
-        with ThreadPoolExecutor(max_workers=len(global_data['model_configs'])) as executor:
-            futures = [executor.submit(self._load_model, config) for config in tqdm(global_data['model_configs'], desc="Loading models")]
-            for future in tqdm(as_completed(futures), total=len(global_data['model_configs']), desc="Loading models complete"):
-                future.result()
+        self.load_models_once()
+
+    def load_models_once(self):
+        if not self.models:  # Load models only once
+            with ThreadPoolExecutor(max_workers=len(global_data['model_configs'])) as executor:
+                futures = [executor.submit(self._load_model, config) for config in tqdm(global_data['model_configs'], desc="Loading models")]
+                for future in tqdm(as_completed(futures), total=len(global_data['model_configs']), desc="Loading models complete"):
+                    future.result()
 
     def _load_model(self, model_config):
         model_name = model_config['name']