Ash2505 commited on
Commit
b634739
·
verified ·
1 Parent(s): 586457a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -9
app.py CHANGED
@@ -128,12 +128,12 @@ def process_image(input_image: Image.Image, fg_threshold: float, mg_threshold: f
128
 
129
  return (
130
  seg_blur,
131
- seg_mask,
132
  depth_map_img,
133
  lens_blur_img,
134
- mask_fg_img,
135
- mask_mg_img,
136
- mask_bg_img
137
  )
138
 
139
  title = "Blur Effects: Gaussian & Depth-Based Lens Blur with Adjustable Depth Thresholds"
@@ -153,12 +153,8 @@ demo = gr.Interface(
153
  ],
154
  outputs=[
155
  gr.Image(type="pil", label="Segmentation-Based Blur"),
156
- gr.Image(type="pil", label="Segmentation Mask"),
157
  gr.Image(type="pil", label="Depth Map"),
158
- gr.Image(type="pil", label="Depth-Based Lens Blur"),
159
- gr.Image(type="pil", label="Foreground Depth Mask"),
160
- gr.Image(type="pil", label="Middleground Depth Mask"),
161
- gr.Image(type="pil", label="Background Depth Mask")
162
  ],
163
  title=title,
164
  description=description,
 
128
 
129
  return (
130
  seg_blur,
131
+ # seg_mask,
132
  depth_map_img,
133
  lens_blur_img,
134
+ # mask_fg_img,
135
+ # mask_mg_img,
136
+ # mask_bg_img
137
  )
138
 
139
  title = "Blur Effects: Gaussian & Depth-Based Lens Blur with Adjustable Depth Thresholds"
 
153
  ],
154
  outputs=[
155
  gr.Image(type="pil", label="Segmentation-Based Blur"),
 
156
  gr.Image(type="pil", label="Depth Map"),
157
+ gr.Image(type="pil", label="Depth-Based Lens Blur")
 
 
 
158
  ],
159
  title=title,
160
  description=description,