Update app.py

app.py

@@ -39,5 +39,6 @@ with gr.Blocks(theme=theme) as demo:
     textbox = gr.Textbox(show_label=False, placeholder='type your prompt in here')
     button = gr.Button("generate",  variant="primary")
     button.click(fn=generate, inputs=textbox, outputs=img)
+    textbox.submit(fn=generate, inputs=textbox, outputs=img)
 
 demo.launch()