Update app.py

app.py

@@ -20,21 +20,20 @@ model_list = [
 ]
 
 default_model = "stabilityai/stable-diffusion-2"
-model_name = gr.inputs.Dropdown(model_list, label="Select Model", default=default_model)
+model_name = gr.outputs.Dropdown(model_list, label="Select Model", default=default_model)
 model = None
 
 def load_model(model_name):
     global model
-    model = gr.Interface.load(model_name, api_key=api_key)
+    model = gr.load(model_name, api_key=api_key)
 
 def predict(inputs):
     return model.predict(inputs)
 
 iface = gr.Interface(
     fn=predict,
-    inputs=model,
+    inputs=model_name,
     outputs="text",
-    capture_session=True,
     title="Model Selection App",
     description="Choose a model and input data to make predictions."
 )