mc_resolution

app.py

@@ -69,9 +69,9 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
     return image
 
 
-def generate(image):
+def generate(image, mc_resolution):
     scene_codes = model(image, device=device)
-    mesh = model.extract_mesh(scene_codes)[0]
+    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)
@@ -109,6 +109,13 @@ with gr.Blocks() as demo:
                         value=0.85,
                         step=0.05,
                     )
+                    mc_resolution = gr.Slider(
+                        label="Marching Cubes Resolution",
+                        minimum=32,
+                        maximum=320,
+                        value=256,
+                        step=32
+                     )
             with gr.Row():
                 submit = gr.Button("Generate", elem_id="generate", variant="primary")
         with gr.Column():