ash11sh commited on
Commit
0ea018c
Β·
1 Parent(s): 56b4f50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -164,7 +164,7 @@ def imsize(img, max_size=512, maintain_aspect_ratio=True):
164
 
165
 
166
  def rem_glass(input_img):
167
- w_0, h_0 = input_img.size[0],input_img.size[1]
168
  #resizing
169
  im = imsize(input_img, max_size=512, maintain_aspect_ratio=False)
170
 
@@ -203,6 +203,6 @@ def rem_glass(input_img):
203
  return Image.open('output/out.png')
204
 
205
 
206
- demo = gr.Interface(rem_glass, gr.Image(shape=(200, 200)), "image")
207
  if __name__ == "__main__":
208
  demo.launch()
 
164
 
165
 
166
  def rem_glass(input_img):
167
+ w_0, h_0 = input_img.size
168
  #resizing
169
  im = imsize(input_img, max_size=512, maintain_aspect_ratio=False)
170
 
 
203
  return Image.open('output/out.png')
204
 
205
 
206
+ demo = gr.Interface(rem_glass, gr.inputs.Image(type="pil"))
207
  if __name__ == "__main__":
208
  demo.launch()