fix

app.py

@@ -55,8 +55,8 @@ with block:
         print("depth reconstruction")
         image = inputs[raw_image]
         # depth reconstruction
-        mask = inputs[masks][0][0]
-        path = dpt.generate_obj(image, inputs[n_samples], mask)
+        #mask = inputs[masks][0][0]
+        path = dpt.generate_obj(image, inputs[n_samples])
 
         return {pcl_figure: path}
     

inference.py

@@ -70,7 +70,7 @@ class DepthPredictor:
         ax.scatter(points,size=0.01, c=colors, marker='o')
         return fig
     
-    def generate_obj(self, image, n_samples, mask):
+    def generate_obj(self, image, n_samples):
         # Step 1: Create a point cloud
         point_cloud, color_array = self.generate_pcl(image)
         #point_cloud, color_array = point_cloud[mask.ravel()[:-1]], color_array[mask.ravel()[:-1]]