fCola commited on
Commit
ebfaf6c
·
verified ·
1 Parent(s): c918044

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -444,7 +444,7 @@ def send_message(message, history):
444
  #return history
445
  tokenizer = generator.tokenizer
446
  streamer = TextIteratorStreamer(tokenizer, skip_prompt=True)
447
-
448
  gen_kwargs = {
449
  "inputs": input_ids,
450
  "streamer": streamer,
 
444
  #return history
445
  tokenizer = generator.tokenizer
446
  streamer = TextIteratorStreamer(tokenizer, skip_prompt=True)
447
+ input_ids = self.tokenizer.encode(message, return_tensors="pt")
448
  gen_kwargs = {
449
  "inputs": input_ids,
450
  "streamer": streamer,