Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -70,7 +70,7 @@ def chat(message, history, temperature, max_new_tokens):
|
|
70 |
truncation=True,
|
71 |
pad_token_id=tokenizer.eos_token_id,
|
72 |
streamer=streamer # Use the TextStreamer here
|
73 |
-
)
|
74 |
for new_token in streamer:
|
75 |
outputs.append(new_token)
|
76 |
if new_token in stop_tokens:
|
|
|
70 |
truncation=True,
|
71 |
pad_token_id=tokenizer.eos_token_id,
|
72 |
streamer=streamer # Use the TextStreamer here
|
73 |
+
)
|
74 |
for new_token in streamer:
|
75 |
outputs.append(new_token)
|
76 |
if new_token in stop_tokens:
|