Update app.py

app.py

@@ -1,4 +1,5 @@
 from transformers import GLPNFeatureExtractor, GLPNForDepthEstimation
+from transformers import AutoFeatureExtractor, AutoModelForDepthEstimation
 import torch
 import numpy as np
 from PIL import Image
@@ -9,8 +10,11 @@ import os
 # url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 # image = Image.open(requests.get(url, stream=True).raw)
 
-feature_extractor = GLPNFeatureExtractor.from_pretrained("vinvino02/glpn-nyu")
-model = GLPNForDepthEstimation.from_pretrained("vinvino02/glpn-nyu")
+# feature_extractor = GLPNFeatureExtractor.from_pretrained("vinvino02/glpn-nyu")
+# model = GLPNForDepthEstimation.from_pretrained("vinvino02/glpn-nyu")
+
+feature_extractor = AutoFeatureExtractor.from_pretrained("Intel/dpt-large")
+model = AutoModelForDepthEstimation.from_pretrained("Intel/dpt-large")
 
 example_list = [["examples/" + example] for example in os.listdir("examples")]