Update app.py

app.py

@@ -36,7 +36,8 @@ def generate(prompt):
     filename = f'moves-{pgn}.svg'
     with open(filename, 'w') as f:
         f.write(img)
-    return pgn, filename
+    plot = f'<img src="{filename}">'
+    return pgn, plot
 
 
 with gr.Blocks() as demo:
@@ -50,7 +51,7 @@ 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)
+    img = gr.Markdown()
     submit = gr.Button("Submit")
     submit.click(generate, [prompt], [output, img])