refactor: Update generate_mvs function to generate multi-view synthesis with a random sample seed and 75 sample steps

launch/model_generation.py

@@ -76,9 +76,12 @@ def get_render_cameras(batch_size=1, M=120, radius=2.5, elevation=10.0, is_flexi
 
 
 @spaces.GPU
-def generate_mvs(input_image, sample_steps, sample_seed):
+def generate_mvs(input_image):
+    sample_seed = np.random.randint(0, 1000000)
     seed_everything(sample_seed)
 
+    sample_steps = 75
+
     z123_image = pipeline(
         input_image, num_inference_steps=sample_steps).images[0]
 
@@ -137,6 +140,9 @@ def make3d(images):
 
 def model_generation_ui(processed_image):
     with gr.Column():
+        with gr.Row():
+            submit_mesh = gr.Button(
+                "Generate 3D Model", elem_id="generate", variant="primary")
         with gr.Row():
             with gr.Column():
                 mv_show_images = gr.Image(
@@ -145,37 +151,21 @@ def model_generation_ui(processed_image):
                     width=379,
                     interactive=False
                 )
-        with gr.Row():
-            with gr.Group():
-                sample_seed = gr.Number(
-                    value=42, label="Seed Value", precision=0)
-                sample_steps = gr.Slider(
-                    label="Sample Steps", minimum=30, maximum=75, value=75, step=5)
-        with gr.Row():
-            submit_mesh = gr.Button(
-                "Generate 3D Model", elem_id="generate", variant="primary")
-        with gr.Row():
-            with gr.Tab("OBJ"):
-                output_model_obj = gr.Model3D(
-                    label="Output Model (OBJ Format)",
-                    interactive=False,
-                )
-                gr.Markdown(
-                    "Note: Downloaded .obj model will be flipped. Export .glb instead or manually flip it before usage.")
-            with gr.Tab("GLB"):
-                output_model_glb = gr.Model3D(
-                    label="Output Model (GLB Format)",
-                    interactive=False,
-                )
-                gr.Markdown(
-                    "Note: The model shown here has a darker appearance. Download to get correct results.")
-        with gr.Row():
-            gr.Markdown(
-                '''Try a different <b>seed value</b> if the result is unsatisfying (Default: 42).''')
+            with gr.Column():
+                with gr.Tab("OBJ"):
+                    output_model_obj = gr.Model3D(
+                        label="Output Model (OBJ Format)",
+                        interactive=False,
+                    )
+                with gr.Tab("GLB"):
+                    output_model_glb = gr.Model3D(
+                        label="Output Model (GLB Format)",
+                        interactive=False,
+                    )
 
     mv_images = gr.State()
 
-    submit_mesh.click(fn=generate_mvs, inputs=[processed_image, sample_steps, sample_seed], outputs=[mv_images, mv_show_images]).success(
+    submit_mesh.click(fn=generate_mvs, inputs=[processed_image], outputs=[mv_images, mv_show_images]).success(
         fn=make3d, inputs=[mv_images], outputs=[
             output_model_obj, output_model_glb]
     )