Update app.py

app.py

@@ -3,7 +3,7 @@ import time
 import numpy as np
 import cv2
 import torch
-from transformers import DPTFeatureExtractor, DPTForDepthEstimation
+from transformers import AutoImageProcessor, AutoModelForDepthEstimation
 from fastapi import FastAPI, File, UploadFile, Response
 from fastapi.responses import FileResponse, HTMLResponse
 from PIL import Image
@@ -15,8 +15,8 @@ app = FastAPI()
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # 🟢 Tải model DPT-Hybrid để tăng tốc
-feature_extractor = DPTFeatureExtractor.from_pretrained("Intel/dpt-swinv2-tiny-256")
-model = DPTForDepthEstimation.from_pretrained("Intel/dpt-swinv2-tiny-256").to(device)
+feature_extractor = AutoImageProcessor.from_pretrained("Intel/dpt-swinv2-tiny-256")
+model = AutoModelForDepthEstimation.from_pretrained("Intel/dpt-swinv2-tiny-256").to(device)
 model.eval()
 
 @app.post("/analyze_path/")