wp

app.py

@@ -146,8 +146,7 @@ def ply_to_glb(ply_file, glb_file):
 
 def save_ply(pc, file_name, grid_size):
     # Produce a mesh (with vertex colors)
-    with open(file_name, "wb") as f:
-        write_ply(f, pc)
+    pc.save(file_name)
 
 
 def create_gif(pc):
@@ -259,7 +258,6 @@ with block:
         with gr.Column():
             model_gif = gr.Video(label="3D Model GIF")
             # btn_pc_to_obj = gr.Button(value="Convert to OBJ", visible=False)
-            model_3d = gr.Model3D(value=None)
             file_out = gr.File(label="Files", visible=False)
 
     if torch.cuda.is_available():
@@ -269,7 +267,7 @@ with block:
                 ["images/fantasy_world.png"],
             ],
             inputs=[input_image],
-            outputs=[model_3d, model_gif, file_out],
+            outputs=[model_gif, file_out],
             fn=generate_3D,
             cache_examples=True
         )