Update app.py

app.py

@@ -22,12 +22,9 @@ def load_fn(models):
             try:
                 m = gr.load(f'models/{model}')
             except Exception as error:
-                print(f"Error loading model {model}: {error}")
-        return gr.Interface(
-            fn=lambda txt: f"Failed to load model {model}.",
-            inputs=gr.Textbox(label="Input Prompt"),
-            outputs=gr.Textbox(label="Output"),
-        )
+    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())
 
 
 load_fn(models)