Update app.py

app.py

@@ -15,19 +15,33 @@ from point_cloud_generator import display_pcd
 # params
 CANCEL_PROCESSING = False
 
-# Initialize models (but actual loading happens in decorated functions)
+# Initialize models (but don't load them yet)
 img_seg = ImageSegmenter(model_type="yolov8s-seg")
 depth_estimator = MonocularDepthEstimator(model_type="midas_v21_small_256")
 
-@spaces.GPU(duration=30)  # Adjust duration based on your needs
+@spaces.GPU
 def process_image(image):
-    image = utils.resize(image)
-    image_segmentation, objects_data = img_seg.predict(image)
-    depthmap, depth_colormap = depth_estimator.make_prediction(image)
-    dist_image = utils.draw_depth_info(image, depthmap, objects_data)
-    objs_pcd = utils.generate_obj_pcd(depthmap, objects_data)
-    plot_fig = display_pcd(objs_pcd)
-    return image_segmentation, depth_colormap, dist_image, plot_fig
+    try:
+        print("Starting image processing")
+        image = utils.resize(image)
+        print("Image resized")
+        
+        # Models will be loaded here when needed
+        image_segmentation, objects_data = img_seg.predict(image)
+        print("Segmentation complete")
+        
+        depthmap, depth_colormap = depth_estimator.make_prediction(image)
+        print("Depth estimation complete")
+        
+        dist_image = utils.draw_depth_info(image, depthmap, objects_data)
+        objs_pcd = utils.generate_obj_pcd(depthmap, objects_data)
+        plot_fig = display_pcd(objs_pcd)
+        return image_segmentation, depth_colormap, dist_image, plot_fig
+    except Exception as e:
+        print(f"Error in process_image: {str(e)}")
+        import traceback
+        print(traceback.format_exc())
+        raise
 
 @spaces.GPU(duration=30)
 def test_process_img(image):