Update app.py

app.py

@@ -4,15 +4,22 @@ import numpy as np
 from PIL import Image
 import uvicorn
 import cv2
-
+from fastdepth import FastDepth
+model = FastDepth(pretrained=True)
+model.eval()
 app = FastAPI()
 
+def analyzepath(image):
+    depth_map = model(image).squeeze().cpu().numpy()
+    return detect_path(depth_map)  # Xử lý đường đi nhanh hơn
 @app.post("/analyze_path/")
 async def analyze_path(file: UploadFile = File(...)):
     image_bytes = await file.read()
     image = Image.open(io.BytesIO(image_bytes)).convert("L")  # Chuyển ảnh sang grayscale
-    depth_map = np.array(image)
-    flipped_depth_map = cv2.flip(depth_map, -1)
+    mImage = cv2.flip(image, -1)
+    #depth_map = np.array(image)
+    depth_map = analyzepath(mImage)
+    
     # Phân tích ảnh Depth Map
     command = detect_path(flipped_depth_map)