Update app.py

app.py

@@ -21,10 +21,13 @@ def load_fn(models):
     for model in models:
         if model not in models_load.keys():
             try:
-                m = gr.load(f'models/{model}')
+                m = gr.Interface.load(f'models/{model}')
+                models_load[model] = m
+                print(f"Loaded model: {model}")
             except Exception as error:
+                print(f"Error loading model {model}: {error}")
                 m = gr.Interface(lambda txt: None, ['text'], ['image'])
-            models_load.update({model: m})
+                models_load[model] = m
 
 load_fn(models)