Update app.py

app.py

@@ -22,10 +22,10 @@ def load_fn(models):
             try:
                 m = gr.load(f'models/{model}')
             except Exception as error:
-    print(f"Error loading model {model}: {error}")
-    # Fallback to a dummy interface with valid components
-    m = gr.Interface(lambda txt: None, inputs=gr.Textbox(), outputs=gr.Image())
-
+                print(f"Error loading model {model}: {error}")
+                m = gr.Interface(lambda txt: None, inputs=gr.Textbox(), outputs=gr.Image())
+            models_load.update({model: m})
+    
 
 load_fn(models)