Update app.py

app.py

@@ -22,8 +22,13 @@ def load_fn(models):
             try:
                 m = gr.load(f'models/{model}')
             except Exception as error:
-                m = gr.Interface(lambda txt: None, ['text'], ['image'])
-            models_load.update({model: m})
+                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"),
+        )
+
 
 load_fn(models)