glb

app.py

@@ -74,13 +74,15 @@ def generate(image):
     mesh = model.extract_mesh(scene_codes)[0]
     mesh = to_gradio_3d_orientation(mesh)
     mesh_path = tempfile.NamedTemporaryFile(suffix=".obj", delete=False)
+    mesh_path2 = tempfile.NamedTemporaryFile(suffix=".glb", delete=False)
     mesh.export(mesh_path.name)
-    return mesh_path.name
+    mesh.export(mesh_path2.name)
+    return mesh_path.name, mesh_path2.name
 
 def run_example(image_pil):
     preprocessed = preprocess(image_pil, False, 0.9)
-    mesh_name = generate(preprocessed)
-    return preprocessed, mesh_name
+    mesh_name, mesn_name2 = generate(preprocessed)
+    return preprocessed, mesh_name, mesh_name2
 
 with gr.Blocks() as demo:
     gr.Markdown(HEADER)
@@ -110,18 +112,23 @@ with gr.Blocks() as demo:
             with gr.Row():
                 submit = gr.Button("Generate", elem_id="generate", variant="primary")
         with gr.Column():
-            with gr.Tab("Model"):
+            with gr.Tab("obj"):
                 output_model = gr.Model3D(
                     label="Output Model",
                     interactive=False,
                 )
+            with gr.Tab("glb"):
+                output_model2 = gr.Model3D(
+                    label="Output Model",
+                    interactive=False,
+                )
     with gr.Row(variant="panel"):
         gr.Examples(
             examples=[
                 os.path.join("examples", img_name) for img_name in sorted(os.listdir("examples"))
             ],
             inputs=[input_image],
-            outputs=[processed_image, output_model],
+            outputs=[processed_image, output_model, output_model2],
             #cache_examples=True,
             fn=partial(run_example),
             label="Examples",
@@ -134,7 +141,7 @@ with gr.Blocks() as demo:
     ).success(
         fn=generate,
         inputs=[processed_image],
-        outputs=[output_model],
+        outputs=[output_model, output_model2],
     )
 
 demo.queue(max_size=10)