Spaces:

NN-BRD
/

hackathon_depth_segment

Runtime error

s194649 commited on Aug 19, 2023

Commit

5f453af

1 Parent(s): 640f5b4

fixed

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -265,7 +265,7 @@ class SegmentPredictor:
     def encode(self, image):
         image = np.array(image)
-        return self.encode_image(image)
     def cond_pred(self, embedding, pts, lbls):
         lbls = np.array(lbls)

     def encode(self, image):
         image = np.array(image)
+        return self.conditioned_pred.encode_image(image)
     def cond_pred(self, embedding, pts, lbls):
         lbls = np.array(lbls)