Update app.py

app.py

@@ -81,21 +81,19 @@ def process_image_pipeline(image):
     blurred_image = apply_depth_based_blur(image, depth_map)
     gaussian_blur_image = apply_blur(image, segmentation_mask)
     
-    return image, Image.fromarray(segmentation_mask), Image.fromarray((depth_map / np.max(depth_map) * 255).astype(np.uint8)), blurred_image, gaussian_blur_image
+    return Image.fromarray(segmentation_mask), blurred_image, gaussian_blur_image
 
 iface = gr.Interface(
     fn=process_image_pipeline,
     inputs=gr.Image(type="pil"),
     outputs=[
-        gr.Image(label="Original Image"),
-       # gr.Image(label="Segmentation Mask"),
-       # gr.Image(label="Depth Map"),
-        gr.Image(label="Lens Blur Effects"),
-        gr.Image(label="Gaussian Blur Effects")
+        gr.Image(label="Segmentation Mask"),
+        gr.Image(label="Depth-based Blurred Image"),
+        gr.Image(label="Gaussian Blur Image")
     ],
     title="Segmentation and Depth-Based Image Processing",
-    description="Upload an image to get segmentation mask, depth map, depth-based blur effect, and Gaussian blur effect."
+    description="Upload an image to get segmentation mask, depth-based blur effect, and Gaussian blur effect."
 )
 
 if __name__ == "__main__":
-    iface.launch(share=True)
+    iface.launch()