Update app.py

app.py

@@ -25,10 +25,11 @@ def generate(text = ""):
   for new_text in streamer:
     yield generated_text + new_text    
     generated_text += new_text
-    if early_stop_pattern in generated_text:
+    if tok.eos_token in generated_text:
       generated_text = generated_text[: generated_text.find(tok.eos_token) if tok.eos_token else None]
       streamer.end()
       yield generated_text
+      return
   return generated_text
 
 demo = gr.Interface(