Update app.py

app.py

@@ -37,42 +37,24 @@ 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):
-    models_load = {}  # Dictionary to store loaded models
-
+    global models_load
+    models_load = {}
     for model in models:
-        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
+        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})
+                traceback.print_exc()  # Prints full error stack trace for debugging
+                #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
+    #return models_load  # Return dictionary instead of using global
 
 load_fn(models)
 num_models = len(models)