Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-
+import plotly.graph_objects as go
 
 import torch
 from tqdm.auto import tqdm
@@ -45,8 +45,24 @@ def inference(prompt):
         samples = x
     pc = sampler.output_to_point_clouds(samples)[0]
     pc = sampler.output_to_point_clouds(samples)[0]
-    fig = plot_point_cloud(pc, grid_size=2, fixed_bounds=((-0.75, -0.75, -0.75),(0.75, 0.75, 0.75)))
-    return fig 
+    colors=(238, 75, 43)
+    fig = go.Figure(
+        data=[
+            go.Scatter3d(
+                x=pc.coords[:,0], y=pc.coords[:,1], z=pc.coords[:,2], 
+                mode='markers',
+                marker=dict(size=1, color=colors)
+            )
+        ],
+        layout=dict(
+            scene=dict(
+                xaxis=dict(visible=False),
+                yaxis=dict(visible=False),
+                zaxis=dict(visible=False)
+            )
+        )
+    )
+    return fig
 
 demo = gr.Interface(fn=inference, inputs="text", outputs=gr.Plot(), examples=[["a red motorcycle"]])
 demo.launch(debug=True)