teaevo commited on
Commit
7c29a25
·
1 Parent(s): 54210ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -14,7 +14,8 @@ table = pd.DataFrame.from_dict(data)
14
 
15
  def chatbot_response(user_message):
16
 
17
- inputs = tokenizer.encode("User: " + user_message, return_tensors="pt")
 
18
  encoding = tokenizer(table=table, query=inputs, return_tensors="pt")
19
  outputs = model.generate(**encoding)
20
  response = tokenizer.batch_decode(outputs, skip_special_tokens=True)
 
14
 
15
  def chatbot_response(user_message):
16
 
17
+ #inputs = tokenizer.encode("User: " + user_message, return_tensors="pt")
18
+ inputs = user_message
19
  encoding = tokenizer(table=table, query=inputs, return_tensors="pt")
20
  outputs = model.generate(**encoding)
21
  response = tokenizer.batch_decode(outputs, skip_special_tokens=True)