use device in example

README.md

@@ -66,13 +66,15 @@ from PIL import Image
 import torch
 from transformers import DepthProImageProcessorFast, DepthProForDepthEstimation
 
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
 url = 'https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg'
 image = Image.open(requests.get(url, stream=True).raw)
 
 image_processor = DepthProImageProcessorFast.from_pretrained("geetu040/DepthPro")
-model = DepthProForDepthEstimation.from_pretrained("geetu040/DepthPro")
+model = DepthProForDepthEstimation.from_pretrained("geetu040/DepthPro").to(device)
 
-inputs = image_processor(images=image, return_tensors="pt")
+inputs = image_processor(images=image, return_tensors="pt").to(device)
 
 with torch.no_grad():
     outputs = model(**inputs)