Update app.py

app.py

@@ -173,6 +173,15 @@ def load_model():
 # Convert depth map to 3D mesh
 def depth_to_mesh(depth_map, image, resolution=100):
     """Convert depth map to 3D mesh"""
+    # Convert depth_map to numpy array if it's a PIL Image
+    if isinstance(depth_map, Image.Image):
+        depth_map = np.array(depth_map)
+    
+    # Make sure the depth map is 2D
+    if len(depth_map.shape) > 2:
+        # If it's a 3D array (like RGB), convert to grayscale
+        depth_map = np.mean(depth_map, axis=2) if depth_map.shape[2] > 1 else depth_map[:,:,0]
+    
     # Get dimensions
     h, w = depth_map.shape
     
@@ -345,6 +354,8 @@ def convert_image_to_3d():
                         depth_map = depth_map.cpu().numpy()
                     elif hasattr(depth_map, 'numpy'):
                         depth_map = depth_map.numpy()
+                    elif isinstance(depth_map, Image.Image):
+                        depth_map = np.array(depth_map)
                     
                     return depth_map