Update app.py

app.py

@@ -30,10 +30,13 @@ tokenizer = Tokenizer.from_pretrained(tokenizer_path)
 def generate(prompt):
     model_input = torch.tensor(tokenizer.encode(prompt), dtype=torch.long, device=device).view((1, len(prompt)))
     pgn = tokenizer.decode(model.generate(model_input, max_new_tokens=4, context_size=context_size)[0].tolist())
-    pgn = StringIO(pgn)
-    game = chess.pgn.read_game(pgn)
+    pgn_str = StringIO(pgn)
+    game = chess.pgn.read_game(pgn_str)
     img = chess.svg.board(game.board())
-    return pgn, img
+    filename = f'moves-{pgn}'
+    with open(filename, 'w') as f:
+        f.write(img)
+    return pgn, filename
 
 
 with gr.Blocks() as demo: