Spaces:

SHSH0819
/

event_detection_app

Sleeping

SHSH0819 commited on Aug 20, 2023

Commit

c83b960

1 Parent(s): 7011da5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -43,7 +43,7 @@ def predict(data):
     model = DistillBERTClass('distilbert-base-cased')
     #print("================  model init ===========================")
-    pretrained_model=torch.load(model_path + "/event_domain_final.pt",map_location=torch.device('cpu'))
     model.load_state_dict(pretrained_model['model_state_dict'])
     model.to(device)

     model = DistillBERTClass('distilbert-base-cased')
     #print("================  model init ===========================")
+    pretrained_model=torch.load(model_path + "event_domain_final.pt",map_location=torch.device('cpu'))
     model.load_state_dict(pretrained_model['model_state_dict'])
     model.to(device)