ailab commited on
Commit
8f2aa2e
·
1 Parent(s): 9c3a29b

segmentation

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -102,11 +102,13 @@ def run_inference(input_img):
102
 
103
  demo = gr.Interface(
104
  fn=run_inference,
105
- inputs=gr.Image(shape=(400, 600)),
106
  outputs=["plot"],
107
  examples=[
108
- "ADE_val_00000001.jpeg", "ADE_val_00001159.jpg",
109
- "ADE_val_00001248.jpg", "ADE_val_00001472.jpg"
 
 
110
  ],
111
  allow_flagging="never",
112
  )
 
102
 
103
  demo = gr.Interface(
104
  fn=run_inference,
105
+ inputs=gr.Image(type="numpy", label="Input Image"),
106
  outputs=["plot"],
107
  examples=[
108
+ "ADE_val_00000001.jpeg",
109
+ "ADE_val_00001159.jpg",
110
+ "ADE_val_00001248.jpg",
111
+ "ADE_val_00001472.jpg"
112
  ],
113
  allow_flagging="never",
114
  )