Hugging Face's logo

kanav0183
/
toxic_model

Model card Files Community
kanav0183 commited on
Commit
5f4c80b
·
1 Parent(s): e1a5a68

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -82,7 +82,7 @@ class DistilBERTClass(torch.nn.Module):
82
  model = DistilBERTClass()
83
  model.to(DEVICE);
84
 
85
- model_loaded = torch.load('model/inference_models_output_4fold_distilbert_fold_best_model.pth')
86
 
87
  model.load_state_dict(model_loaded['model'])
88
 
 
82
  model = DistilBERTClass()
83
  model.to(DEVICE);
84
 
85
+ model_loaded = torch.load('inference_models_output_4fold_distilbert_fold_best_model.pth')
86
 
87
  model.load_state_dict(model_loaded['model'])
88