nevreal commited on
Commit
9d52d71
·
verified ·
1 Parent(s): f698509

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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()