Update app.py

app.py

@@ -8,6 +8,7 @@ import numpy as np
 import time
 import requests
 import logging
+import traceback  # For better error reporting
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 logging.basicConfig(level=logging.WARNING)
@@ -36,19 +37,42 @@ def sanitize_file_name(file_name, max_length=100):
 
 
     
+#def load_fn(models):
+  #  global models_load
+  #  models_load = {}
+  #  for model in models:
+      #  if model not in models_load.keys():
+      #      try:
+      #          m = gr.load(f'models/{model}')
+      #          print(f"{m}\n");
+      #      except Exception as error:
+        #        print(f"Error loading model {model}: {error}\n")
+       #         m = gr.Interface(lambda _: None, inputs=gr.Textbox(), outputs=gr.Image(), queue=False)
+      #      models_load.update({model: m})
+            
 def load_fn(models):
-    global models_load
-    models_load = {}
+    models_load = {}  # Dictionary to store loaded models
+
     for model in models:
-        if model not in models_load.keys():
-            try:
-                m = gr.load(f'models/{model}')
-                print(f"{m}\n");
-            except Exception as error:
-                print(f"Error loading model {model}: {error}\n")
-                m = gr.Interface(lambda _: None, inputs=gr.Textbox(), outputs=gr.Image(), queue=False)
-            models_load.update({model: m})
+        try:
+            print(f"Attempting to load model: {model}")  # Debugging print
+
+            # Load model properly (replace this with your actual model-loading logic)
+            # If `gr.load()` is incorrect for your use case, use a different loading method.
+            m = gr.Interface.load(f'models/{model}')  # Adjust as needed
+
+            print(f"Successfully loaded model: {model}\n")
+
+        except Exception as error:
+            print(f"Error loading model {model}: {error}")
+            traceback.print_exc()  # Prints full error stack trace for debugging
+
+            # Assign a dummy interface to avoid crashes
+            #m = gr.Interface(fn=lambda _: None, inputs=gr.Textbox(), outputs=gr.Image(), queue=False)
+            models_load[model] = None
+        models_load[model] = m  # Store in dictionary
 
+    return models_load  # Return dictionary instead of using global
 
 load_fn(models)
 num_models = len(models)