Update app.py

app.py

@@ -51,7 +51,10 @@ def inference(prompt):
             go.Scatter3d(
                 x=pc.coords[:,0], y=pc.coords[:,1], z=pc.coords[:,2], 
                 mode='markers',
-                marker=dict(size=1, color=colors)
+                marker=dict(
+                  size=2,
+                  color=['rgb({},{},{})'.format(r,g,b) for r,g,b in zip(pc.channels["R"], pc.channels["G"], pc.channels["B"])],
+              )
             )
         ],
         layout=dict(
@@ -60,7 +63,7 @@ def inference(prompt):
                 yaxis=dict(visible=False),
                 zaxis=dict(visible=False)
             )
-        )
+        ),
     )
     return fig