Update app.py

app.py

@@ -50,10 +50,10 @@ with gr.Blocks() as demo:
     """)
     prompt = gr.Text(label="PGN")
     output = gr.Text(label="Next turn", interactive=False)
-
+    img = gr.Image(width=500, height=500)
     submit = gr.Button("Submit")
-    submit.click(generate, [prompt], [output])
-    img = gr.Image()
+    submit.click(generate, [prompt], [output, img])
+    
     gr.Examples(
         [
             ["1. e4", ],