Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from depth_map_generator import generate_depth_map
 from depth_segmentation import segment_image_by_depth
 
-def process_image(image, num_segments):
+def on_process(image, num_segments):
     # Save the uploaded image
     original_img_path = "original.jpg"
     image.save(original_img_path)
@@ -13,6 +13,7 @@ def process_image(image, num_segments):
     # Perform depth segmentation
     segmented_images = segment_image_by_depth(original_img_path, depth_map_path, num_segments)
     
+    # Return the segmented images and file paths for download
     return segmented_images, segmented_images
 
 # Gradio Interface