fix2

app.py

@@ -6,18 +6,21 @@ import os, glob
 import spaces
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
+
 model_name = 'hyunseoki/ReMoDetect-deberta'
 
 THESHOLD=4.0
-predictor = AutoModelForSequenceClassification.from_pretrained(model_name)
+predictor = AutoModelForSequenceClassification.from_pretrained(model_name, force_download=True)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
+predictor.eval()
 
 @spaces.GPU
 def predict(text):
     device = "cuda" if torch.cuda.is_available() else "cpu"
     predictor.to(device)
     tokenized = tokenizer(text, return_tensors='pt', truncation=True, max_length=512).to(device)
-    result = predictor(**tokenized).logits[0].cpu().detach().item()
+    with torch.no_grad():
+        result = predictor(**tokenized).logits[0].cpu().detach().item()
     AI_score = round(torch.sigmoid(torch.tensor(result-THESHOLD)*2).item(),2)
     return f'{AI_score*100} %', f'{round(result,2)}'