Update predictors.py

predictors.py

@@ -153,6 +153,15 @@ def predict_quillbot(text):
         return q_score
 
 
+def predict_proba_quillbot(text):
+    with torch.no_grad():
+        tokenized_text = quillbot_tokenizer(texts, return_tensors="pt", padding=True).to(device)
+        outputs = quillbot_model(**tokenized_text)
+        tensor_logits = outputs[0]
+        probas = F.softmax(tensor_logits).detach().cpu().numpy()
+    return probas
+
+
 def predict_bc(model, tokenizer, text):
     with torch.no_grad():
         model.eval()