Update app.py

app.py

@@ -23,7 +23,7 @@ model_list = [
 default_model = "stabilityai/stable-diffusion-2"
 model_name = gr.inputs.Dropdown(choices=model_list, label="Select Model", default=default_model)
 
-def generate_image(text, model_name):
+def generate_image(text, default_model):
     model = gr.load(default_model, source="huggingface", api_key=api_key)
     return model.predict(text)