Spaces:

EuroPython2022
/

ToxicCommentClassification

Runtime error

ShAnSantosh commited on Jul 16, 2022

Commit

eb844ee

1 Parent(s): 72a0576

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -93,7 +93,7 @@ def predict(comment=None) -> dict:
     model = ToxicModel()
-    model.load_state_dict(torch.load("toxic_model_0.pth", map_location=torch.device(device)))
     model.to(device)
     predicted = inference_fn(model, ids, mask, token_type_ids)

     model = ToxicModel()
+    model.load_state_dict(torch.load("toxicx_model_0.pth", map_location=torch.device(device)))
     model.to(device)
     predicted = inference_fn(model, ids, mask, token_type_ids)