Update app.py

app.py

@@ -184,7 +184,10 @@ def main():
                 result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
                 result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
 
-        run_button.click(fn=run,inputs=[prompt,*result],outputs=result)
+        gr.on(
+            triggers=[run_button.click, prompt.submit],
+            fn=run,inputs=[prompt,*result],outputs=result
+        )
         demo.queue().launch()
 
 if __name__ == "__main__":