Update app.py

app.py

@@ -174,10 +174,6 @@ def extract_gaussian(state: dict, req: gr.Request) -> Tuple[str, str]:
 
 output_buf = gr.State()
 video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
-model_output = gr.Model3D(label="Extracted GLB/Gaussian", height=300)
-
-# Add a hidden JSON output for the state object for API calls
-state_output_json = gr.JSON(visible=False, label="State JSON Output")
 
 with gr.Blocks(delete_cache=(600, 600)) as demo:
     gr.Markdown("""
@@ -236,8 +232,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
     ).then(
         text_to_3d,
         inputs=[text_prompt, seed, ss_guidance_strength, ss_sampling_steps, slat_guidance_strength, slat_sampling_steps],
-        # Output state to hidden JSON first, then video to visible component, then state to internal buffer
-        outputs=[state_output_json, video_output, output_buf],
+        outputs=[output_buf, video_output],
     ).then(
         lambda: tuple([gr.Button(interactive=True), gr.Button(interactive=True)]),
         outputs=[extract_glb_btn, extract_gs_btn],
@@ -276,24 +271,4 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
 if __name__ == "__main__":
     pipeline = TrellisTextTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-text-xlarge")
     pipeline.cuda()
-    demo.launch()
-
-# --- API-only endpoint for server integration ---
-# This exposes text_to_3d with gr.JSON() as the first output, so the state object is included in the API response.
-# Not wired to the UI; use for API calls only.
-api_text_to_3d = gr.Interface(
-    fn=lambda prompt, seed, ss_guidance_strength, ss_sampling_steps, slat_guidance_strength, slat_sampling_steps: text_to_3d(
-        prompt, seed, ss_guidance_strength, ss_sampling_steps, slat_guidance_strength, slat_sampling_steps, gr.Request()
-    ),
-    inputs=[
-        gr.Textbox(label="Text Prompt"),
-        gr.Number(label="Seed"),
-        gr.Number(label="SS Guidance Strength"),
-        gr.Number(label="SS Sampling Steps"),
-        gr.Number(label="SLAT Guidance Strength"),
-        gr.Number(label="SLAT Sampling Steps"),
-    ],
-    outputs=[gr.JSON(label="State Object"), gr.Textbox(label="Video Path")],
-    allow_flagging="never",
-    description="API endpoint for text_to_3d that returns the state object as JSON. Not for UI use.",
-)
+    demo.launch()