Update app.py

app.py

@@ -14,8 +14,8 @@ app = FastAPI()
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # 🟢 Tải model DPT-Hybrid thay cho ZoeDepth để tăng tốc
-feature_extractor = DPTFeatureExtractor.from_pretrained("Intel/dpt-hybrid-midas")
-model = DPTForDepthEstimation.from_pretrained("Intel/dpt-hybrid-midas").to(device)
+feature_extractor = DPTFeatureExtractor.from_pretrained("Intel/dpt-small")
+model = DPTForDepthEstimation.from_pretrained("Intel/dpt-small").to(device)
 model.eval()
 
 @app.post("/analyze_path/")
@@ -27,7 +27,7 @@ async def analyze_path(file: UploadFile = File(...)):
     image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
 
     # 🔵 Resize ảnh để xử lý nhanh hơn
-    image = image.resize((256, 256))  # Giảm kích thước giúp tăng tốc độ xử lý
+    image = image.resize((192, 192))  # Giảm kích thước giúp tăng tốc độ xử lý
     image_np = np.array(image)
     flipped_image = cv2.flip(image_np, -1)
     # 🟢 Chuẩn bị ảnh cho mô hình