Chaerin5 commited on
Commit
e774d53
·
1 Parent(s): 8bcba28

runtime fix

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -189,7 +189,7 @@ if NEW_MODEL:
189
  model_path = './DINO_EMA_11M_b50_lr1e-5_epoch7_step380k.ckpt'
190
  elif MODEL_EPOCH == 6:
191
  # model_path = "./DINO_EMA_11M_b50_lr1e-5_epoch6_step320k.ckpt"
192
- model_path = hf_hub_download(repo_id="Chaerin5/FoundHand-weights", filename="DINO_EMA_11M_b50_lr1e-5_epoch6_step320k.ckpt", token=True)
193
  elif MODEL_EPOCH == 4:
194
  model_path = "./DINO_EMA_11M_b50_lr1e-5_epoch4_step210k.ckpt"
195
  elif MODEL_EPOCH == 10:
 
189
  model_path = './DINO_EMA_11M_b50_lr1e-5_epoch7_step380k.ckpt'
190
  elif MODEL_EPOCH == 6:
191
  # model_path = "./DINO_EMA_11M_b50_lr1e-5_epoch6_step320k.ckpt"
192
+ model_path = hf_hub_download(repo_id="Chaerin5/FoundHand-weights", filename="DINO_EMA_11M_b50_lr1e-5_epoch6_step320k.ckpt")
193
  elif MODEL_EPOCH == 4:
194
  model_path = "./DINO_EMA_11M_b50_lr1e-5_epoch4_step210k.ckpt"
195
  elif MODEL_EPOCH == 10: