0708_8

app.py

@@ -20,7 +20,7 @@ base_model = base_model
 tokenizer = AutoTokenizer.from_pretrained(base_model)
 
 sen_model = Classifier(base_model, num_labels=2, device=device, tokenizer=tokenizer)
-sen_model.load_state_dict(torch.load('/resolve/main/'+sen_model_file))
+sen_model.load_state_dict(torch.load(sen_model_file))
 
 entity_model = Classifier(base_model, num_labels=2, device=device, tokenizer=tokenizer)
 entity_model.load_state_dict(torch.load(entity_model_file))