Spaces:

NN-BRD
/

hackathon_depth_segment

Runtime error

jens commited on Aug 1, 2023

Commit

2ba4f1e

1 Parent(s): 67eca52

fix

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -16,7 +16,7 @@ class DepthPredictor:
     def predict(self, image):
         # prepare image for the model
-        inputs = self.processor(images=image, return_tensors="pt").to(self.device)
         with torch.no_grad():
             outputs = self.model(**inputs)
             predicted_depth = outputs.predicted_depth

     def predict(self, image):
         # prepare image for the model
+        inputs = self.processor(images=image, return_tensors="pt")
         with torch.no_grad():
             outputs = self.model(**inputs)
             predicted_depth = outputs.predicted_depth