Spaces:

iamkhadke
/

GeneralChatBot

Runtime error

iamkhadke commited on Apr 22, 2023

Commit

b1a0ae4

•

1 Parent(s): bf0b168

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -53,7 +53,7 @@ def chat(curr_system_message, history):
     # Tokenize the messages string
     model_inputs = tok([messages], return_tensors="pt")
     streamer = TextIteratorStreamer(
-        tok, timeout=10., skip_prompt=True, skip_special_tokens=True)
     generate_kwargs = dict(
         model_inputs,
         streamer=streamer,

     # Tokenize the messages string
     model_inputs = tok([messages], return_tensors="pt")
     streamer = TextIteratorStreamer(
+        tok, skip_prompt=True, skip_special_tokens=True)
     generate_kwargs = dict(
         model_inputs,
         streamer=streamer,