Update app.py

app.py

@@ -25,7 +25,7 @@ model = None
 
 def load_model(model_name):
     global model
-    model = gr.load(model_name, api_key=api_key)
+    model = gr.load(model_name, source="huggingface", api_key=api_key)
 
 def predict(inputs):
     return model.predict(inputs)
@@ -40,4 +40,4 @@ iface = gr.Interface(
 
 load_model(default_model)  # Load the default model initially
 
-iface.launch()
+iface.launch()