Hasanmog commited on
Commit
6a4d941
·
verified ·
1 Parent(s): 0c5c1cc
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ login(token=os.getenv("HUGGINGFACE_HUB_TOKEN"))
38
  # Use this command for evaluating the Grounding DINO model
39
  config_file = "cfg_odvg.py"
40
  ckpt_repo_id = "Hasanmog/Peft-GroundingDINO"
41
- ckpt_filename = "Best.pth"
42
 
43
  def load_model_hf(model_config_path, repo_id, filename, device='cpu'):
44
  args = SLConfig.fromfile(model_config_path)
 
38
  # Use this command for evaluating the Grounding DINO model
39
  config_file = "cfg_odvg.py"
40
  ckpt_repo_id = "Hasanmog/Peft-GroundingDINO"
41
+ ckpt_filename = "checkpoint.pth"
42
 
43
  def load_model_hf(model_config_path, repo_id, filename, device='cpu'):
44
  args = SLConfig.fromfile(model_config_path)