tonyassi commited on
Commit
2383f2b
·
verified ·
1 Parent(s): 84ce883

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -9,7 +9,16 @@ def detect(img, threshold):
9
  img_with_boxes = foot_detection.draw_boxes(img)
10
  return img_with_boxes
11
 
12
- demo = gr.Interface(fn=detect,
 
 
 
 
 
 
 
 
 
13
  inputs=[gr.Image(label='Input', type='pil'), gr.Slider(label='Threshold', minimum=0.0, maximum=1.0, value=0.3)],
14
  outputs=gr.Image(label='Result', type='pil'))
15
  demo.launch()
 
9
  img_with_boxes = foot_detection.draw_boxes(img)
10
  return img_with_boxes
11
 
12
+ demo = gr.Interface(fn=detect,
13
+ title='Foot Detection',
14
+ description="""by [Tony Assi](https://www.tonyassi.com/)
15
+
16
+ A lightweight Python module for detecting feet or shoes in images using a fine-tuned Faster R-CNN model (PyTorch + Torchvision).
17
+
18
+ [![Model](https://img.shields.io/badge/%F0%9F%A4%97-Models-yellow)](https://huggingface.co/tonyassi/foot-detection)
19
+
20
+ [![Github](https://img.shields.io/badge/Github-Repo-blue)](https://github.com/TonyAssi/foot-detection)
21
+ """,
22
  inputs=[gr.Image(label='Input', type='pil'), gr.Slider(label='Threshold', minimum=0.0, maximum=1.0, value=0.3)],
23
  outputs=gr.Image(label='Result', type='pil'))
24
  demo.launch()