Update app.py

app.py

@@ -1,5 +1,5 @@
 import torch
-from transformers import pipeline, TextStreamer
+from transformers import pipeline, TextIteratorStreamer
 import gradio as gr
 import threading
 import time
@@ -57,7 +57,7 @@ def chat(message, history, temperature, max_new_tokens):
     
     # Create a TextStreamer for token streaming
     tokenizer = model_pipeline.tokenizer
-    streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
+    streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)