Update binoculars/detector.py

binoculars/detector.py

@@ -96,8 +96,7 @@ class Binoculars(object):
                         encodings.to(DEVICE_1), self.tokenizer.pad_token_id)
         binoculars_scores = ppl / x_ppl
         binoculars_scores = binoculars_scores.tolist()
-        # return binoculars_scores[0] if isinstance(input_text, str) else binoculars_scores
-        return binoculars_scores
+        return binoculars_scores[0] if isinstance(input_text, str) else binoculars_scores
 
     def predict(self, input_text: Union[list[str], str]) -> Union[list[str], str]:
         # binoculars_scores = np.array(self.compute_score(input_text))