Spaces:

jflo
/

text_classification

Sleeping

jflo commited on 8 days ago

Commit

ef5a40f

1 Parent(s): 2e6e7c9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -84,7 +84,7 @@ def predict(request: PredictRequest):
         num_soreness_levels=3
     )
-    model.load_state_dict(torch.load('best_DistilBERTmodel.pt', map_location=torch.device('cpu')))
     model.to(device)
     model.eval()

         num_soreness_levels=3
     )
+    model.load_state_dict(torch.load('best_DistilBERT_model.pt', map_location=torch.device('cpu')))
     model.to(device)
     model.eval()