fix

app.py

@@ -53,7 +53,7 @@ with block:
         print("depth reconstruction")
         image = inputs[raw_image]
         # depth reconstruction
-        fig = dpt.generate_fig(image)
+        fig = dpt.generate_fig2(image)
         return {pcl_figure: fig}
     
     depth_reconstruction_btn.click(on_depth_reconstruction_btn_click, components, [pcl_figure], queue=False)

inference.py

@@ -9,6 +9,7 @@ import requests
 import open3d as o3d
 import pandas as pd
 import plotly.express as px
+import matplotlib.pyplot as plt
 
 class DepthPredictor:
     def __init__(self):
@@ -60,6 +61,14 @@ class DepthPredictor:
         # Step 6: Create a 3D scatter plot using Plotly Express
         fig = px.scatter_3d(df, x='x', y='y', z='z', color='red', size=size)
         return fig
+     
+    def generate_fig2(self, image):
+        points, colors = self.generate_pcl(image)
+        # Step 6: Create a 3D scatter plot using Plotly Express
+        fig = plt.figure()
+        ax = fig.add_subplot(111, projection='3d')
+        ax.scatter(points,size=0.01, c=colors, marker='o')
+        return fig