Spaces:

rusen
/

gpt_detector

Running

rusen commited on Jan 9, 2024

Commit

e01d8e6

1 Parent(s): 303d729

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -10,7 +10,7 @@ def model_predict(model, tokenizer, sentences):
     """
-    inputs = tokenizer(sentences, padding=True, truncation=True, return_tensors="pt", max_length=512).to(device)
     # Classify sentences
     with torch.no_grad():
         outputs = model(**inputs) # get the logits

     """
+    inputs = tokenizer(sentences, padding=True, truncation=True, return_tensors="pt", max_length=512).to("cpu")
     # Classify sentences
     with torch.no_grad():
         outputs = model(**inputs) # get the logits