Spaces:

rusen
/

gpt_detector

Running

rusen commited on Jan 9, 2024

Commit

dbe1544

1 Parent(s): 74510a3

corrected typo

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -19,4 +19,4 @@ def model_predict(model, tokenizer, sentences):
         outputs = model(**inputs) # get the logits
         label = np.argmax(outputs.logits.to("cpu"))
-    return int(labels)

         outputs = model(**inputs) # get the logits
         label = np.argmax(outputs.logits.to("cpu"))
+    return int(label)