Update app.py

app.py

@@ -18,13 +18,18 @@ if not os.path.exists("fastdepth"):
 # 🟢 Thêm `fastdepth` vào `sys.path`
 sys.path.append(os.path.abspath("fastdepth"))
 
+# 🟢 Đảm bảo file trọng số tồn tại
+weights_path = "fastdepth/models/fastdepth_nyu.pt"
+if not os.path.exists(weights_path):
+    os.system(f"wget -O {weights_path} https://github.com/dwofk/fast-depth/raw/master/models/fastdepth_nyu.pt")
+
 # 🟢 Import FastDepth
-from fastdepth.models import MobileNetSkipAdd  # Model chính của FastDepth
+from fastdepth.models import MobileNetSkipAdd  
 
 # 🟢 Load mô hình FastDepth
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = MobileNetSkipAdd(output_size=(224, 224))  # 🔹 Thêm output_size
-model.load_state_dict(torch.load("fastdepth/models/fastdepth_nyu.pt", map_location=device))
+model = MobileNetSkipAdd(output_size=(224, 224))  # 🔹 Fix lỗi thiếu `output_size`
+model.load_state_dict(torch.load(weights_path, map_location=device))
 model.eval().to(device)
 
 @app.post("/analyze_path/")