Update app.py

app.py

@@ -37,7 +37,7 @@ 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, (128, 128))
+    image = cv2.resize(image, (256, 144))
     image = torch.from_numpy(image).permute(2, 0, 1).unsqueeze(0).float().to(device)
     return image / 255.0
 
@@ -46,7 +46,7 @@ def plot_depth_map(depth_map):
     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(0, 1)
+    ax.set_zlim(1, 0)
     plt.close(fig)
     return fig