Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from io import StringIO
 from model import DecoderTransformer, Tokenizer
 from huggingface_hub import hf_hub_download
 import torch
@@ -29,6 +30,7 @@ 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)
     img = chess.svg.board(game.board())
     return pgn, img