Update README.md

README.md

@@ -65,16 +65,16 @@ import numpy as np
 import requests
 import torch
 
-from transformers import DPTForDepthEstimation, DPTFeatureExtractor
+from transformers import DPTImageProcessor, DPTForDepthEstimation
 
+image_processor = DPTImageProcessor.from_pretrained("Intel/dpt-hybrid-midas")
 model = DPTForDepthEstimation.from_pretrained("Intel/dpt-hybrid-midas", low_cpu_mem_usage=True)
-feature_extractor = DPTFeatureExtractor.from_pretrained("Intel/dpt-hybrid-midas")
 
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
 
 # prepare image for the model
-inputs = feature_extractor(images=image, return_tensors="pt")
+inputs = image_processor(images=image, return_tensors="pt")
 
 with torch.no_grad():
     outputs = model(**inputs)