jioji0 commited on
Commit
61c3f37
1 Parent(s): 7e6a221
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -97,13 +97,13 @@ def sepia(input_img):
97
  fig = draw_plot(pred_img, seg)
98
  return fig
99
 
100
- def proc_img(img):
101
- img = img.resize((1024,1024))
102
- return img
103
 
104
  demo = gr.Interface(fn=sepia,
105
- #inputs=gr.Image(shape=(1024, 1024)),
106
- inputs=gr.Image(type='pil', prepeocessing_function=proc_img),
107
  outputs=['plot'],
108
  examples=["city-1c.jpg"],
109
  allow_flagging='never')
 
97
  fig = draw_plot(pred_img, seg)
98
  return fig
99
 
100
+ #def prepro_img(img):
101
+ # img = img.resize((1024,1024))
102
+ # return img
103
 
104
  demo = gr.Interface(fn=sepia,
105
+ inputs=gr.Image(shape=(400, 600)),
106
+ #inputs=gr.Image(type='pil', prepeocessing_function=prepro_img),
107
  outputs=['plot'],
108
  examples=["city-1c.jpg"],
109
  allow_flagging='never')