Update app.py

app.py

@@ -31,7 +31,7 @@ async def analyze_path(file: UploadFile = File(...)):
     # 🔵 Resize ảnh để xử lý nhanh hơn
     image = image.resize((256, 256))
     image_np = np.array(image)
-   // flipped_image = cv2.flip(image_np, -1)  # Lật ảnh
+   # flipped_image = cv2.flip(image_np, -1)  # Lật ảnh
 
     # 🟢 Chuẩn bị ảnh cho mô hình
     inputs = processor(images=image_np, return_tensors="pt").to(device)
@@ -42,8 +42,8 @@ async def analyze_path(file: UploadFile = File(...)):
 
     # 🟢 Xử lý ảnh sau khi dự đoán
     predicted_depth = outputs.predicted_depth.squeeze().cpu().numpy()
-    depth_map = (predicted_depth * 255 / predicted_depth.max()).astype("uint8")
-
+    depth_map = (predicted_depth - predicted_depth.min()) / (predicted_depth.max() - predicted_depth.min())
+    #depth_map = (predicted_depth * 255 / predicted_depth.max()).astype("uint8")
     # 🔵 Chuyển depth_map thành ảnh có thể hiển thị
     depth_colored = cv2.applyColorMap(depth_map, cv2.COLORMAP_INFERNO)
     depth_pil = Image.fromarray(depth_colored)