yunnn426 commited on
Commit
b363393
1 Parent(s): 321db28

delete fig2

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -116,15 +116,14 @@ def sepia(input_img):
116
  pred_img = np.array(input_img) * 0.5 + color_seg * 0.5
117
  pred_img = pred_img.astype(np.uint8)
118
 
119
- fig = draw_plot(pred_img, seg),
120
- fig2 = input_img
121
- return fig, fig2
122
 
123
  demo = gr.Interface(fn=sepia,
124
  title="🚥Sidewalk Segmentation🏃‍️",
125
  description="Image Segmentation for Sidewalks",
126
  inputs=gr.Image(),
127
- outputs=['plot', 'plot2'],
128
  examples=["sidewalk_1.jpg", "sidewalk_2.jpg", "sidewalk_3.jpg", "sidewalk_4.jpg", "sidewalk_5.jpg"],
129
  allow_flagging='never',
130
  theme="gradio/soft",
 
116
  pred_img = np.array(input_img) * 0.5 + color_seg * 0.5
117
  pred_img = pred_img.astype(np.uint8)
118
 
119
+ fig = draw_plot(pred_img, seg)
120
+ return fig
 
121
 
122
  demo = gr.Interface(fn=sepia,
123
  title="🚥Sidewalk Segmentation🏃‍️",
124
  description="Image Segmentation for Sidewalks",
125
  inputs=gr.Image(),
126
+ outputs=['plot'],
127
  examples=["sidewalk_1.jpg", "sidewalk_2.jpg", "sidewalk_3.jpg", "sidewalk_4.jpg", "sidewalk_5.jpg"],
128
  allow_flagging='never',
129
  theme="gradio/soft",