anilbhatt1 commited on
Commit
5a46991
1 Parent(s): a12f9e6

Modified app.py to align description & added example input images to be displayed on interface

Browse files
Files changed (3) hide show
  1. app.py +9 -4
  2. broadway_tower_rgb.jpeg +0 -0
  3. jeep.jpeg +0 -0
app.py CHANGED
@@ -37,13 +37,18 @@ def gradio_fn(pil_input_img):
37
  np_img_array = np.array(pil_image_output)
38
  return np_img_array
39
 
 
 
 
 
40
  demo = gr.Interface(fn=gradio_fn,
41
- inputs=gr.Image(type="pil"),
42
  outputs="image",
43
  title="FAST-SAM Segment Everything",
44
  description="- **FastSAM** model that returns segmented RGB image of given input image. \
45
- - **Credits** : \
46
- - https://huggingface.co/An-619 \
47
- - https://github.com/CASIA-IVA-Lab/FastSAM")
 
48
 
49
  demo.launch(share=True)
 
37
  np_img_array = np.array(pil_image_output)
38
  return np_img_array
39
 
40
+ example1 = './broadway_tower_rgb.jpeg'
41
+ example2 = '/jeep.jpeg'
42
+ examples = [[example1, 0.5, -1], [example2, 0.5, -1]]
43
+
44
  demo = gr.Interface(fn=gradio_fn,
45
+ inputs=[gr.Image(type="pil",label="Input Image")],
46
  outputs="image",
47
  title="FAST-SAM Segment Everything",
48
  description="- **FastSAM** model that returns segmented RGB image of given input image. \
49
+ **Credits** : \
50
+ https://huggingface.co/An-619 & \
51
+ https://github.com/CASIA-IVA-Lab/FastSAM",
52
+ examples=examples)
53
 
54
  demo.launch(share=True)
broadway_tower_rgb.jpeg ADDED
jeep.jpeg ADDED