upperwal commited on
Commit
3f8ce4f
·
verified ·
1 Parent(s): 8dd81fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ def generate_response(transcription):
59
 
60
  print(tokenized_chat)
61
 
62
- tokenized_chat = tokenizer(tokenized_chat)
63
 
64
  input_ids = tokenized_chat['input_ids'].to(device)
65
  if len(input_ids.shape) == 1:
 
59
 
60
  print(tokenized_chat)
61
 
62
+ tokenized_chat = tokenizer(tokenized_chat, return_tensors="pt")
63
 
64
  input_ids = tokenized_chat['input_ids'].to(device)
65
  if len(input_ids.shape) == 1: