0708_9

app.py

@@ -3,7 +3,6 @@ from models import *
 from huggingface_hub import hf_hub_download
 import os
 from config import *
-device = torch.device('cuda' if torch.cuda.is_available() else "cpu")
 
 ENTITY_REPO_ID = 'vaivTA/absa_v2_entity'
 ENTITY_FILENAME = "entity_model.pt"
@@ -19,18 +18,18 @@ base_model = base_model
 
 tokenizer = AutoTokenizer.from_pretrained(base_model)
 
-sen_model = Classifier(base_model, num_labels=2, device=device, tokenizer=tokenizer)
+sen_model = Classifier(base_model, num_labels=2, device='cpu', tokenizer=tokenizer)
 sen_model.load_state_dict(torch.load(sen_model_file))
 
-entity_model = Classifier(base_model, num_labels=2, device=device, tokenizer=tokenizer)
+entity_model = Classifier(base_model, num_labels=2, device='cpu', tokenizer=tokenizer)
 entity_model.load_state_dict(torch.load(entity_model_file))
 
 
 def infer(test_sentence):
-    entity_model.to(device)
-    entity_model.eval()
-    sen_model.to(device)
-    sen_model.eval()
+    # entity_model.to(device)
+    # entity_model.eval()
+    # sen_model.to(device)
+    # sen_model.eval()
     
     form = test_sentence
     annotation = []