fix

app.py

@@ -35,8 +35,9 @@ with block:
             input_image = gr.Image(label='Input', height=512, type='pil')
             masks_annotated_image = gr.AnnotatedImage(label='Segments', height=512)
             pcl_figure = gr.Model3D(label="3-D Reconstruction", clear_color=[1.0, 1.0, 1.0, 1.0])
+            n_samples = gr.Slider(minimum=1e3, maximum=1e6, step=1e3, default=1e3, label='Number of Samples')
             #gr.Plot(label='3D Reconstruction')
-            #cutout_galary = gr.Gallery(label='Cutouts', object_fit='contain', height=512)
+            #cutout_galary = gr.Gallery(label='Cutouts', object_fit='contain', height=512)  
         with gr.Row():
             with gr.Column(scale=1):
                 with gr.Row():
@@ -49,12 +50,12 @@ with block:
     # components
     components = {point_coords, point_labels, raw_image, masks, cutout_idx, input_image,
                   point_label_radio, text, reset_btn, sam_sgmt_everything_btn,
-                  sam_decode_btn, depth_reconstruction_btn, masks_annotated_image}
+                  sam_decode_btn, depth_reconstruction_btn, masks_annotated_image, n_samples}
     def on_depth_reconstruction_btn_click(inputs):
         print("depth reconstruction")
         image = inputs[raw_image]
         # depth reconstruction
-        path = dpt.generate_obj(image)
+        path = dpt.generate_obj(image, inputs[n_samples])
         return {pcl_figure: path}
     
     depth_reconstruction_btn.click(on_depth_reconstruction_btn_click, components, [pcl_figure], queue=False)

inference.py

@@ -70,11 +70,11 @@ class DepthPredictor:
         ax.scatter(points,size=0.01, c=colors, marker='o')
         return fig
     
-    def generate_obj(self, image):
+    def generate_obj(self, image, n_samples):
         # Step 1: Create a point cloud
         point_cloud, color_array = self.generate_pcl(image)
         # sample 1000 points
-        idxs = np.random.choice(len(point_cloud), int(1e6))
+        idxs = np.random.choice(len(point_cloud), int(n_samples))
         point_cloud = point_cloud[idxs]
         color_array = color_array[idxs]
         # Create a mesh to hold the colored cubes