xinghaochen commited on
Commit
de08842
1 Parent(s): bd42c45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -232,10 +232,10 @@ def segment_with_box(
232
  return image, session_state
233
 
234
 
235
- img_p = gr.Image(label="Input with points", type="pil")
236
- img_b = gr.Image(label="Input with box", type="pil")
237
 
238
- with gr.Blocks(css=css, title="EdgeSAM") as demo:
239
  session_state = gr.State({
240
  'coord_list': [],
241
  'label_list': [],
@@ -307,7 +307,7 @@ with gr.Blocks(css=css, title="EdgeSAM") as demo:
307
  with gr.Row():
308
  with gr.Column(scale=1):
309
  gr.Markdown(
310
- "<center><img src='https://visitor-badge.laobi.icu/badge?page_id=chongzhou/edgesam' alt='visitors'></center>")
311
 
312
  img_p.upload(on_image_upload, [img_p, session_state], [img_p, session_state])
313
  img_p.select(segment_with_points, [add_or_remove, session_state], [img_p, session_state])
 
232
  return image, session_state
233
 
234
 
235
+ img_p = gr.Image(label="Input with points", type="pil", interactive=True)
236
+ img_b = gr.Image(label="Input with box", type="pil", interactive=True)
237
 
238
+ with gr.Blocks(css=css, title="TinySAM") as demo:
239
  session_state = gr.State({
240
  'coord_list': [],
241
  'label_list': [],
 
307
  with gr.Row():
308
  with gr.Column(scale=1):
309
  gr.Markdown(
310
+ "<center><img src='https://visitor-badge.laobi.icu/badge?page_id=xinghaochen.tinysam' alt='visitors'></center>")
311
 
312
  img_p.upload(on_image_upload, [img_p, session_state], [img_p, session_state])
313
  img_p.select(segment_with_points, [add_or_remove, session_state], [img_p, session_state])