, resolution=mc_resolution

app.py

@@ -17,7 +17,6 @@ from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orienta
 
 HEADER = """
 **TripoSR** is a state-of-the-art open-source model for **fast** feedforward 3D reconstruction from a single image, developed in collaboration between [Tripo AI](https://www.tripo3d.ai/) and [Stability AI](https://stability.ai/).
-
 **Tips:**
 1. If you find the result is unsatisfied, please try to change the foreground ratio. It might improve the results.
 2. Please disable "Remove Background" option only if your input image is RGBA with transparent background, image contents are centered and occupy more than 70% of image width or height.
@@ -69,16 +68,15 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
     return image
 
 
-def generate(image, mc_resolution, formats=["obj", "glb"]):
+def generate(image, mc_resolution):
     scene_codes = model(image, device=device)
     mesh = model.extract_mesh(scene_codes, resolution=mc_resolution)[0]
     mesh = to_gradio_3d_orientation(mesh)
-    rv = []
-    for format in formats:
-        mesh_path = tempfile.NamedTemporaryFile(suffix=f".{format}", delete=False)
-        mesh.export(mesh_path.name)
-        rv.append(mesh_path.name)
-    return rv
+    mesh_path = tempfile.NamedTemporaryFile(suffix=".obj", delete=False)
+    mesh_path2 = tempfile.NamedTemporaryFile(suffix=".glb", delete=False)
+    mesh.export(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)
@@ -121,12 +119,12 @@ with gr.Blocks() as demo:
                 submit = gr.Button("Generate", elem_id="generate", variant="primary")
         with gr.Column():
             with gr.Tab("obj"):
-                output_model_obj = gr.Model3D(
+                output_model = gr.Model3D(
                     label="Output Model",
                     interactive=False,
                 )
             with gr.Tab("glb"):
-                output_model_glb = gr.Model3D(
+                output_model2 = gr.Model3D(
                     label="Output Model",
                     interactive=False,
                 )
@@ -136,7 +134,7 @@ with gr.Blocks() as demo:
                 os.path.join("examples", img_name) for img_name in sorted(os.listdir("examples"))
             ],
             inputs=[input_image],
-            outputs=[processed_image, output_model_obj, output_model_glb],
+            outputs=[processed_image, output_model, output_model2],
             #cache_examples=True,
             fn=partial(run_example),
             label="Examples",
@@ -149,8 +147,8 @@ with gr.Blocks() as demo:
     ).success(
         fn=generate,
         inputs=[processed_image, mc_resolution],
-        outputs=[output_model_obj, output_model_glb],
+        outputs=[output_model, output_model2],
     )
 
 demo.queue(max_size=10)
-demo.launch()
+demo.launch()