slslslrhfem commited on
Commit
6da214d
·
1 Parent(s): 508344f

change checkpoints

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,8 +17,8 @@ def download_models_from_hub():
17
  model_dir.mkdir(exist_ok=True)
18
 
19
  models = {
20
- "main": "EmbeddingModel_MERT_768-epoch=0073-val_loss=0.1058-val_acc=0.9585-val_f1=0.9366-val_precision=0.9936-val_recall=0.8857.ckpt",
21
- "backup": "step=007000-val_loss=0.1831-val_acc=0.9278.ckpt"
22
  }
23
 
24
  downloaded_models = {}
 
17
  model_dir.mkdir(exist_ok=True)
18
 
19
  models = {
20
+ "main": "EmbeddingModel_MERT_768_2class_weighted-epoch=0014-val_loss=0.0099-val_acc=0.9993-val_f1=0.9978-val_precision=0.9967-val_recall=0.9989.ckpt",
21
+ "backup": "step=003432-val_loss=0.0216-val_acc=0.9963.ckpt"
22
  }
23
 
24
  downloaded_models = {}