kanav0183
/

toxic_model

Model card Files Files and versions

kanav0183 commited on Oct 27, 2022

Commit

5f4c80b

·

1 Parent(s): e1a5a68

Update app.py

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -82,7 +82,7 @@ class DistilBERTClass(torch.nn.Module):
 model = DistilBERTClass()
 model.to(DEVICE);
-model_loaded = torch.load('model/inference_models_output_4fold_distilbert_fold_best_model.pth')
 model.load_state_dict(model_loaded['model'])

 model = DistilBERTClass()
 model.to(DEVICE);
+model_loaded = torch.load('inference_models_output_4fold_distilbert_fold_best_model.pth')
 model.load_state_dict(model_loaded['model'])