Update app.py
Browse files
app.py
CHANGED
@@ -36,7 +36,7 @@ with gr.Blocks() as demo:
|
|
36 |
generate_button = gr.Button("Generate Image")
|
37 |
|
38 |
# When the button is clicked, call the generate_image function
|
39 |
-
generate_button.click(fn=generate_image, inputs=base_model, main_model, prompt, outputs=output_image)
|
40 |
|
41 |
# Launch the interface
|
42 |
demo.launch()
|
|
|
36 |
generate_button = gr.Button("Generate Image")
|
37 |
|
38 |
# When the button is clicked, call the generate_image function
|
39 |
+
generate_button.click(fn=generate_image, inputs=[base_model, main_model, prompt], outputs=output_image)
|
40 |
|
41 |
# Launch the interface
|
42 |
demo.launch()
|