Update app.py

app.py

@@ -42,7 +42,7 @@ with demo:
     button = gr.components.Button(label="Generate")
     # when the button is clicked, call the generate_image_predictions function
     # and pass in the prompt as an argument
-    button.submit(generate_image_predictions, inputs=input, outputs=output)
+    button.click(generate_image_predictions, inputs=input, outputs=output)
 
 
 demo.launch()