fix hard-coded cuda device

model.py

@@ -10,14 +10,15 @@ checkpoint = "geetu040/DepthPro"
 revision = "project"
 image_processor = DepthProImageProcessorFast.from_pretrained(checkpoint, revision=revision)
 model = DepthProForDepthEstimation.from_pretrained(checkpoint, revision=revision)
-model = model.to('cuda')
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model = model.to(device)
 
 def predict(image):
 	# inference
 
 	# prepare image for the model
 	inputs = image_processor(images=image, return_tensors="pt")
-	inputs = {k: v.to('cuda') for k, v in inputs.items()}
+	inputs = {k: v.to(device) for k, v in inputs.items()}
 
 	with torch.no_grad():
 		outputs = model(**inputs)