Update app.py

app.py

@@ -19,6 +19,8 @@ from trellis.pipelines import TrellisImageTo3DPipeline
 from trellis.representations import Gaussian, MeshExtractResult
 from trellis.utils import render_utils, postprocessing_utils
 
+NUM_INFERENCE_STEPS = 8
+
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
 # Constants
 MAX_SEED = np.iinfo(np.int32).max
@@ -91,7 +93,6 @@ def generate_flux_image(
     width: int,
     height: int,
     guidance_scale: float,
-    num_inference_steps: int,
     progress: gr.Progress = gr.Progress(track_tqdm=True),
 ) -> Image.Image:
     """Generate image using Flux pipeline"""
@@ -102,7 +103,7 @@ def generate_flux_image(
     image = flux_pipeline(
         prompt=prompt,
         guidance_scale=guidance_scale,
-        num_inference_steps=num_inference_steps,
+        num_inference_steps=NUM_INFERENCE_STEPS,
         width=width,
         height=height,
         generator=generator,
@@ -196,7 +197,7 @@ with gr.Blocks() as demo:
                     height = gr.Slider(512, 1024, label="Height", value=768, step=16)
                 with gr.Row():
                     guidance_scale = gr.Slider(0.0, 10.0, label="Guidance Scale", value=3.5, step=0.1)
-                    num_inference_steps = gr.Slider(1, 50, label="Steps", value=8, step=1)
+                #    num_inference_steps = gr.Slider(1, 50, label="Steps", value=8, step=1)
 
             with gr.Accordion("3D Generation Settings", open=False):
                 gr.Markdown("Stage 1: Sparse Structure Generation")
@@ -238,7 +239,7 @@ with gr.Blocks() as demo:
 
     generate_btn.click(
         generate_flux_image,
-        inputs=[prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+        inputs=[prompt, seed, randomize_seed, width, height, guidance_scale],
         outputs=[generated_image],
     ).then(
         image_to_3d,