aixsatoshi commited on
Commit
2e5208e
1 Parent(s): 248b38e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def stream_chat(message: str, history: list, temperature: float, max_new_tokens:
60
  streamer = TextIteratorStreamer(tokenizer, timeout=10., skip_prompt=True, skip_special_tokens=True)
61
 
62
  generate_kwargs = dict(
63
- inputs,
64
  streamer=streamer,
65
  top_k=top_k,
66
  top_p=top_p,
 
60
  streamer = TextIteratorStreamer(tokenizer, timeout=10., skip_prompt=True, skip_special_tokens=True)
61
 
62
  generate_kwargs = dict(
63
+ inputs['input_ids'],
64
  streamer=streamer,
65
  top_k=top_k,
66
  top_p=top_p,