Update app.py

app.py

@@ -10,6 +10,22 @@ import plotly.graph_objects as go
 import plotly.express as px
 import open3d as o3d
 from depth_anything_v2.dpt import DepthAnythingV2
+import os
+import gdown
+
+
+
+# Define path and file ID
+checkpoint_dir = "checkpoints"
+os.makedirs(checkpoint_dir, exist_ok=True)
+
+model_file = os.path.join(checkpoint_dir, "depth_anything_v2_vitl.pth")
+gdrive_url = "https://drive.google.com/uc?id=141Mhq2jonkUBcVBnNqNSeyIZYtH5l4K5"
+
+# Download if not already present
+if not os.path.exists(model_file):
+    print("Downloading model from Google Drive...")
+    gdown.download(gdrive_url, model_file, quiet=False)
 
 css = """
 #img-display-container {
@@ -41,7 +57,7 @@ model_configs = {
 }
 encoder = 'vitl'
 model = DepthAnythingV2(**model_configs[encoder])
-state_dict = torch.load(f'checkpoints/depth_anything_v2_{encoder}.pth', map_location="cpu")
+state_dict = torch.load(f'depth_anything_v2_{encoder}.pth', map_location="cpu")
 model.load_state_dict(state_dict)
 model = model.to(DEVICE).eval()