Append token to text for generator

app.py

@@ -11,10 +11,10 @@ def generate_text(context, num_of_tokens, temperature=1.0):
         idx = torch.zeros((1,1), dtype=torch.long)
     else:
         idx = torch.tensor(encode(context), dtype=torch.long).unsqueeze(0)
-    
+    text = ""
     for token in model.generate(idx, max_new_tokens=num_of_tokens,temperature=temperature):
-        yield token
-    
+        text+= token
+        yield text
 
 
 with gr.Blocks() as demo: