eljanmahammadli commited on
Commit
104f341
1 Parent(s): b632a34

Update predictors.py

Browse files
Files changed (1) hide show
  1. predictors.py +1 -1
predictors.py CHANGED
@@ -155,7 +155,7 @@ def predict_quillbot(text):
155
 
156
  def predict_proba_quillbot(text):
157
  with torch.no_grad():
158
- tokenized_text = quillbot_tokenizer(texts, return_tensors="pt", padding=True).to(device)
159
  outputs = quillbot_model(**tokenized_text)
160
  tensor_logits = outputs[0]
161
  probas = F.softmax(tensor_logits).detach().cpu().numpy()
 
155
 
156
  def predict_proba_quillbot(text):
157
  with torch.no_grad():
158
+ tokenized_text = quillbot_tokenizer(text, return_tensors="pt", padding=True).to(device)
159
  outputs = quillbot_model(**tokenized_text)
160
  tensor_logits = outputs[0]
161
  probas = F.softmax(tensor_logits).detach().cpu().numpy()