wip

app.py

@@ -116,11 +116,11 @@ def generate_3D(input, model_name='base1B', guidance_scale=3.0, grid_size=128):
     pc = sampler.output_to_point_clouds(samples)[0]
 
     set_state('Converting to mesh...')
-    save_ply(pc, '/tmp/mesh.ply', grid_size)
+    save_ply(pc, '/tmp/mesh.ply')
 
     set_state('')
 
-    return ply_to_glb('/tmp/mesh.ply', '/tmp/mesh.glb'), create_gif(pc), gr.update(value=['/tmp/mesh.glb', '/tmp/mesh.ply'], visible=True)
+    return pc, create_gif(pc), gr.update(value=['/tmp/mesh.ply'], visible=True)
 
 def prepare_img(img):
 
@@ -257,6 +257,7 @@ with block:
 
         with gr.Column():
             model_gif = gr.Video(label="3D Model GIF")
+            model_3d = gr.Model3D(value=None)
             # btn_pc_to_obj = gr.Button(value="Convert to OBJ", visible=False)
             file_out = gr.File(label="Files", visible=False)
 
@@ -267,7 +268,7 @@ with block:
                 ["images/fantasy_world.png"],
             ],
             inputs=[input_image],
-            outputs=[model_gif, file_out],
+            outputs=[model_3d, model_gif, file_out],
             fn=generate_3D,
             cache_examples=True
         )