Spaces:

NN-BRD
/

hackathon_depth_segment

Runtime error

jens commited on Aug 1, 2023

Commit

1d564b4

1 Parent(s): 2ba4f1e

fix

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -10,8 +10,8 @@ import requests
 class DepthPredictor:
     def __init__(self):
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-        self.processor = DPTImageProcessor.from_pretrained("Intel/dpt-large").to(self.device)
-        self.model = DPTForDepthEstimation.from_pretrained("Intel/dpt-large").to(self.device)
         self.model.eval()
     def predict(self, image):

 class DepthPredictor:
     def __init__(self):
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        self.processor = DPTImageProcessor.from_pretrained("Intel/dpt-large")
+        self.model = DPTForDepthEstimation.from_pretrained("Intel/dpt-large")
         self.model.eval()
     def predict(self, image):