Update app.py

app.py

@@ -1,5 +1,6 @@
 import io
 import os
+import sys
 import time
 import numpy as np
 import cv2
@@ -11,17 +12,21 @@ import uvicorn
 
 app = FastAPI()
 
-# 🟢 Cài đặt ZoeDepth từ Hugging Face (chạy trên lần đầu)
-if not os.path.exists("ZoeDepth"):
+# 🟢 Kiểm tra nếu ZoeDepth chưa được tải, thì tải về
+zoe_path = "ZoeDepth"
+if not os.path.exists(zoe_path):
     os.system("git clone https://github.com/isl-org/ZoeDepth.git")
 
-# 🟢 Import ZoeDepth sau khi clone
-from ZoeDepth.zoedepth.models.builder import build_model
-from ZoeDepth.zoedepth.utils.config import get_config
+# 🟢 Thêm ZoeDepth vào sys.path để import được
+sys.path.append(os.path.abspath(zoe_path))
+
+# 🟢 Import ZoeDepth sau khi đã tải về
+from zoedepth.models.builder import build_model
+from zoedepth.utils.config import get_config
 
 # 🟢 Load mô hình ZoeDepth-Tiny
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-config = get_config("ZoeDepth/configs/zoedepth_tiny.py")  # 🟢 Chọn model tiny để chạy nhanh hơn
+config = get_config(f"{zoe_path}/configs/zoedepth_tiny.py")  # 🟢 Chọn model tiny để chạy nhanh hơn
 model = build_model(config).to(device)
 model.eval()