Shanat commited on
Commit
f91bbe8
·
verified ·
1 Parent(s): 4099d28

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ response_list = []
31
 
32
 
33
  def vanilla_chatbot(message, history):
34
- inputs = tokenizer(message['text'], return_tensors="pt").to("cpu")
35
  with torch.no_grad():
36
  outputs = model.generate(inputs.input_ids, max_length=100)
37
  return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
31
 
32
 
33
  def vanilla_chatbot(message, history):
34
+ inputs = tokenizer(message, return_tensors="pt").to("cpu")
35
  with torch.no_grad():
36
  outputs = model.generate(inputs.input_ids, max_length=100)
37
  return tokenizer.decode(outputs[0], skip_special_tokens=True)