Spaces:
Running
Running
eljanmahammadli
commited on
Commit
•
104f341
1
Parent(s):
b632a34
Update predictors.py
Browse files- 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(
|
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()
|