Update app.py

app.py

@@ -20,7 +20,7 @@ def predict_v1(title, abstract):
             model_path_v1,
             num_labels=1,
             load_in_8bit=True,
-        ).to(device)
+        ).eval()
         tokenizer_v1 = AutoTokenizer.from_pretrained(model_path_v1)
         model_v1.eval()
 
@@ -48,7 +48,7 @@ class PaperScorer:
             model_path,
             num_labels=1,
             load_in_8bit=True,
-        ).to(self.device).eval()
+        ).eval()
 
         self.tokenizer = AutoTokenizer.from_pretrained(model_path)
         self.tokenizer.pad_token = self.tokenizer.eos_token