Update app.py

app.py

@@ -7,9 +7,10 @@ model3 = gr.load("models/runwayml/stable-diffusion-v1-5")
 model4 = gr.load("models/stablediffusionapi/juggernaut-xl-v5")
 model5 = gr.load("models/stabilityai/stable-diffusion-xl-base-1.0")
 
-# Function to switch between models
-def generate_image(model, text_input):
-    return model(text_input)
+# Function to switch between models and generate image
+def generate_image(selected_model_name, text_input):
+    selected_model = models[selected_model_name]
+    return selected_model(text_input)
 
 # Define the models and their names
 models = {
@@ -29,16 +30,9 @@ input_text = gr.Textbox(label="Input Text", placeholder="Enter text here")  # Se
 # Create the output image
 output_image = gr.Image()
 
-# Create a function that will be called when the "Generate" button is clicked
-def generate_button_click():
-    selected_model_name = model_dropdown.value
-    selected_model = models[selected_model_name]
-    result_image = generate_image(selected_model, input_text.value)
-    output_image.image(result_image)
-
 # Create the interface
 iface = gr.Interface(
-    
+    fn=generate_image,
     inputs=[model_dropdown, input_text],
     outputs=output_image,
 )