jwlarocque commited on
Commit
d68042c
·
1 Parent(s): 51352c6

Actually fix model paths

Browse files
Files changed (2) hide show
  1. app.py +3 -2
  2. images/{stairs.jpg → bicycle.jpg} +2 -2
app.py CHANGED
@@ -62,7 +62,7 @@ def show_box(box, ax):
62
  ax.add_patch(plt.Rectangle((x0, y0), w, h, edgecolor='green', facecolor=(0,0,0,0), lw=2))
63
 
64
 
65
- sam_checkpoint = hf_hub_download(repo_id="andzhang01/segment-anything", filename="sam_vit_l_0b3195.pth")
66
  # sam_checkpoint = r"~/.cache/huggingface/hub/models--andzhang01--segment-anything/sam_vit_l_0b3195.pth"
67
  model_type = "vit_l"
68
 
@@ -236,7 +236,8 @@ gradio_app = gr.Interface(
236
  ["./images/wire_shelf.jpg", "20,100,480,660"],
237
  ["./images/radio_telescope.jpg", "1130,320,4000,2920"],
238
  ["./images/bridge.jpg", ""],
239
- ["./images/tree.jpg", "70,110,2290,1800"]
 
240
  ]
241
  )
242
 
 
62
  ax.add_patch(plt.Rectangle((x0, y0), w, h, edgecolor='green', facecolor=(0,0,0,0), lw=2))
63
 
64
 
65
+ sam_checkpoint = hf_hub_download(repo_id="andzhang01/segment_anything", filename="sam_vit_l_0b3195.pth")
66
  # sam_checkpoint = r"~/.cache/huggingface/hub/models--andzhang01--segment-anything/sam_vit_l_0b3195.pth"
67
  model_type = "vit_l"
68
 
 
236
  ["./images/wire_shelf.jpg", "20,100,480,660"],
237
  ["./images/radio_telescope.jpg", "1130,320,4000,2920"],
238
  ["./images/bridge.jpg", ""],
239
+ ["./images/tree.jpg", "70,110,2290,1800"],
240
+ ["./images/bicycle.jpg", "135,235,2425,1580"]
241
  ]
242
  )
243
 
images/{stairs.jpg → bicycle.jpg} RENAMED
File without changes