Update gradio_app.py

gradio_app.py

@@ -142,6 +142,7 @@ def image2mesh(view_front: np.ndarray,
                more: bool = False,
                scheluder_name: str ="DDIMScheduler",
                guidance_scale: int = 7.5,
+               steps: int = 50, 
                seed: int = 4,
                octree_depth: int = 7):
     
@@ -160,6 +161,7 @@ def image2mesh(view_front: np.ndarray,
         sample_times=1,
         guidance_scale=guidance_scale,
         return_intermediates=False,
+        steps=steps,
         seed=seed
        
     )[0]
@@ -351,7 +353,7 @@ if __name__=="__main__":
                             outputs=[view_front, view_right, view_back, view_left]
                     ).success(
                             fn=image2mesh, 
-                            inputs=[view_front, view_right, view_back, view_left, more, scheduler, guidance_scale, seed, octree_depth],
+                            inputs=[view_front, view_right, view_back, view_left, more, scheduler, guidance_scale, steps, seed, octree_depth],
                             outputs=outputs, 
                             api_name="generate_img2obj")
         run_mv_btn.click(fn=gen_mvimg,
@@ -359,7 +361,7 @@ if __name__=="__main__":
                         outputs=[view_front, view_right, view_back, view_left]
         )
         run_3d_btn.click(fn=image2mesh, 
-                        inputs=[view_front, view_right, view_back, view_left, more, scheduler, guidance_scale, seed, octree_depth],
+                        inputs=[view_front, view_right, view_back, view_left, more, scheduler, guidance_scale, steps, seed, octree_depth],
                         outputs=outputs, 
                         api_name="generate_img2obj")