, mc_resolution

app.py

@@ -69,15 +69,16 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
     return image
 
 
-def generate(image, mc_resolution):
+def generate(image, mc_resolution, formats=["obj", "glb"]):
     scene_codes = model(image, device=device)
     mesh = model.extract_mesh(scene_codes, resolution=mc_resolution)[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)
-    mesh.export(mesh_path2.name)
-    return mesh_path.name, mesh_path2.name
+    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
 
 def run_example(image_pil):
     preprocessed = preprocess(image_pil, False, 0.9)
@@ -112,7 +113,7 @@ with gr.Blocks() as demo:
                     mc_resolution = gr.Slider(
                         label="Marching Cubes Resolution",
                         minimum=32,
-                        maximum=320,
+                        maximum=1024,
                         value=256,
                         step=32
                      )
@@ -147,7 +148,7 @@ with gr.Blocks() as demo:
         outputs=[processed_image],
     ).success(
         fn=generate,
-        inputs=[processed_image],
+        inputs=[processed_image, mc_resolution],
         outputs=[output_model, output_model2],
     )