Update app.py

app.py

@@ -45,14 +45,17 @@ def load_fn(models):
             try:
                 m = gr.load(f'models/{model}')
                 print(f"{m}\n");
+                models_load.update({model: m})
+                models_load[model] = m  # Store in dictionary
             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})
+                #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