mmdetection

Runtime error

kmn5409 commited on May 2, 2023

Commit

bdcf038

1 Parent(s): a02a022

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -34,7 +34,7 @@ def _update_model_dict_if_hf_token_is_given(
     for dic in model_dict.values():
         ckpt_path = dic['model']
         name = ckpt_path.split('/')[-1]
-        ckpt_path = huggingface_hub.hf_hub_download('kmn5409/mmdetection',
                                                     f'models/{name}',
                                                     use_auth_token=token)
         dic['model'] = ckpt_path

     for dic in model_dict.values():
         ckpt_path = dic['model']
         name = ckpt_path.split('/')[-1]
+        ckpt_path = huggingface_hub.hf_hub_download('kmn5409/Flower_Model',
                                                     f'models/{name}',
                                                     use_auth_token=token)
         dic['model'] = ckpt_path