Soumic commited on
Commit
19c8c1e
·
1 Parent(s): a0b4d21

:hammer_and_pick: Repair a silly mistake

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -480,7 +480,8 @@ def start_bert(classifier_model, criterion, m_optimizer=torch.optim.Adam, WINDOW
480
  )
481
 
482
  # reload
483
- classifier_model = classifier_model.from_pretrained(f"my-awesome-model-{WINDOW}")
 
484
 
485
  pass
486
 
 
480
  )
481
 
482
  # reload
483
+ # classifier_model = classifier_model.from_pretrained(f"fahimfarhan/{classifier_model.model_repository_name}")
484
+ # classifier_model = classifier_model.from_pretrained(model_subdirectory)
485
 
486
  pass
487