Update app.py

app.py

@@ -35,7 +35,7 @@ def generate(prompt, chat_history, max_new_tokens):
   final_prompt += "Output:"
 
   # Streamer
-  streamer = TextIteratorStreamer(tokenizer=tokenizer, skip_prompt=True)
+  streamer = TextIteratorStreamer(tokenizer=tokenizer, skip_prompt=True, skip_special_tokens=True, timeout=300.0)
   thread = Thread(target=phi2, kwargs={"text_inputs":final_prompt, "max_new_tokens":max_new_tokens, "streamer":streamer})
   thread.start()