Update app.py

app.py

@@ -37,16 +37,17 @@ color_map = cv2.applyColorMap(np.arange(256, dtype=np.uint8), cv2.COLORMAP_INFER
 color_map = torch.from_numpy(color_map).to(device)
 
 def preprocess_image(image):
-    image = cv2.resize(image, (256, 144))
+    image = cv2.resize(image, (128, 72))
     image = torch.from_numpy(image).permute(2, 0, 1).unsqueeze(0).float().to(device)
     return image / 255.0
 
 def plot_depth_map(depth_map):
-    fig = plt.figure()
+    fig = plt.figure(figsize=(16, 9))  # Set figure size to 16:9
     ax = fig.add_subplot(111, projection='3d')
     x, y = np.meshgrid(range(depth_map.shape[1]), range(depth_map.shape[0]))
     ax.plot_surface(x, y, depth_map, cmap='viridis')
-    ax.set_zlim(1, 0)
+    ax.view_init(azim=90, elev=0)  # Rotate the view to face forward
+    ax.set_zlim(0, 1)
     plt.close(fig)
     return fig