fixed bool maybe

app.py

@@ -34,7 +34,7 @@ with block:
     with gr.Column():
         with gr.Row():
             with gr.Column():
-                input_image = gr.Image(label='Input', type='pil', tool=None)
+                input_image = gr.Image(label='Input', type='pil', tool=None) # mirror_webcam = False
                 sam_encode_btn = gr.Button('Encode', variant='primary')
                 sam_encode_status = gr.Label('Not encoded yet')
             masks_annotated_image = gr.AnnotatedImage(label='Segments', height=512)

inference.py

@@ -98,10 +98,10 @@ class DepthPredictor:
         # Create cubes and add them to the mesh
         cs = [(255,0,0),(0,255,0),(0,0,255)]
         for c,(mask, _) in zip(cs, masks):
-            if len(mask) == len(point_cloud):
-                mask = mask.ravel()
-            else:
-                mask = mask.ravel()[:-1]
+            #if len(mask) == len(point_cloud):
+            #    mask = mask.ravel()
+            #else:
+            #    mask = mask.ravel()[:-1]
             mask = mask.ravel()
             point_cloud_subset, color_array_subset = point_cloud[mask], color_array[mask]
             idxs = np.random.choice(len(point_cloud_subset), int(n_samples))