idkWhatToUse commited on
Commit
50862fc
·
verified ·
1 Parent(s): 248f534

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -29,15 +29,17 @@ def lookup_item(name):
29
  # 2. 載入模型:GroundingDINO (從你的 Dataset Repo)
30
  # =========================================================
31
  # ⚠️ 請把這裡的 repo_id 換成你的 HuggingFace Dataset
32
- REPO_ID = "idkWhatToUse/groundingdino-weights" # ← 修改這行
33
 
34
  config_path = hf_hub_download(
35
  repo_id=REPO_ID,
36
- filename="GroundingDINO_SwinT_OGC.py"
 
37
  )
38
  checkpoint_path = hf_hub_download(
39
  repo_id=REPO_ID,
40
- filename="groundingdino_swint_ogc.pth"
 
41
  )
42
 
43
  from groundingdino.util.inference import load_model, load_image, predict, annotate
 
29
  # 2. 載入模型:GroundingDINO (從你的 Dataset Repo)
30
  # =========================================================
31
  # ⚠️ 請把這裡的 repo_id 換成你的 HuggingFace Dataset
32
+ REPO_ID = "idkWhatToUse/groundingdino-weights"
33
 
34
  config_path = hf_hub_download(
35
  repo_id=REPO_ID,
36
+ filename="GroundingDINO_SwinT_OGC.py",
37
+ repo_type="dataset"
38
  )
39
  checkpoint_path = hf_hub_download(
40
  repo_id=REPO_ID,
41
+ filename="groundingdino_swint_ogc.pth",
42
+ repo_type="dataset"
43
  )
44
 
45
  from groundingdino.util.inference import load_model, load_image, predict, annotate