Update app.py

app.py

@@ -7,6 +7,7 @@ from fastapi import FastAPI, File, UploadFile
 from PIL import Image
 import uvicorn
 from torchvision import transforms
+from midas.model_loader import load_model  # Thư viện MiDaS
 
 # 🟢 Tạo FastAPI
 app = FastAPI()
@@ -14,9 +15,9 @@ app = FastAPI()
 # 🟢 Kiểm tra GPU
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-# 🟢 Tải model MiDaS
-midas = torch.hub.load("isl-org/MiDaS", "DPT_Swin2_L_384")
-midas.to(device)
+# 🟢 Tải model MiDaS (DPT-Swin2 Large 384)
+model_path = "dpt_swin2_large_384.pt"  # Đảm bảo đã tải file này từ GitHub
+midas = load_model(model_path, device)
 midas.eval()
 
 # 🟢 Chuẩn bị bộ tiền xử lý ảnh