Update app.py

app.py

@@ -1,4 +1,4 @@
-from transformers import DPTImageProcessor, DPTForDepthEstimation
+from transformers import AutoImageProcessor, DPTForDepthEstimation
 import torch
 import numpy as np
 from PIL import Image
@@ -7,11 +7,11 @@ import requests
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
 
-processor = DPTImageProcessor.from_pretrained("Intel/dpt-large")
-model = DPTForDepthEstimation.from_pretrained("Intel/dpt-large")
+image_processor = AutoImageProcessor.from_pretrained("facebook/dpt-dinov2-small-nyu")
+model = DPTForDepthEstimation.from_pretrained("facebook/dpt-dinov2-small-nyu")
 
 # prepare image for the model
-inputs = processor(images=image, return_tensors="pt")
+inputs = image_processor(images=image, return_tensors="pt")
 
 with torch.no_grad():
     outputs = model(**inputs)