Update app.py

app.py

@@ -49,8 +49,9 @@ def plot_depth_map(depth_map, original_image):
     # Rotate by 90 degrees
     x, y = y, x
     
-    # Use original image colors
-    colors = original_image.reshape(-1, 3) / 255.0
+    # Resize original image to match depth map dimensions
+    original_image_resized = cv2.resize(original_image, (depth_map.shape[1], depth_map.shape[0]))
+    colors = original_image_resized.reshape(-1, 3) / 255.0
     
     ax.plot_surface(x, y, depth_map, facecolors=colors, shade=False)
     ax.view_init(elev=70, azim=0)  # Tilt by 20 degrees (90 - 20 = 70)
@@ -74,7 +75,7 @@ def process_frame(image):
     
     depth_map = (depth_map - depth_map.min()) / (depth_map.max() - depth_map.min())
     
-    return plot_depth_map(depth_map, cv2.resize(image, (64, 36)))
+    return plot_depth_map(depth_map, image)  # Pass the original image, not resized
 
 interface = gr.Interface(
     fn=process_frame,