lasty try

app.py

@@ -1,11 +1,9 @@
-# Version: 1.1.0 - API State Fix + DEBUG (Video Disabled - Corrected Baseline) (2025-05-04)
+# Version: 1.1.0 - API State Fix + DEBUG (Video Disabled - FINAL CORRECTED BASELINE) (2025-05-04)
 # Changes:
-# - Based *EXACTLY* on user-provided Version 1.1.0 code.
+# - Based *EXACTLY* on user-provided Version 1.1.0 code that fixed state dict & loaded pipeline.
 # - TEMPORARY DEBUGGING STEP: Commented out video rendering/saving in `text_to_3d`
 #   and return None for video_path to isolate the "Session not found" error.
-# - All other code (imports, functions, UI bindings, pipeline loading) is from Version 1.1.0.
-# - Removed incorrect `torch_dtype` argument from pipeline loading.
-# - Removed incorrect `inputs`/`outputs` arguments from `demo.unload()`.
+# - No other changes were made. Removed previous erroneous additions.
 
 import gradio as gr
 # NOTE: Ensuring 'spaces' is imported if decorators are used (was missing in user provided snippet but needed)
@@ -16,7 +14,7 @@ import spaces
 import os
 import shutil
 os.environ['TOKENIZERS_PARALLELISM'] = 'true'
-os.environ['SPCONV_ALGO'] = 'native' # Direct set as per original
+os.environ['SPCONV_ALGO'] = 'native' # Direct set as per original user code
 
 from typing import *
 import torch
@@ -383,6 +381,7 @@ with gr.Blocks(delete_cache=(600, 600), title="TRELLIS Text-to-3D") as demo:
                         *NOTE: Gaussian file (.ply) can be very large (~50MB+) and may take time to process/download.*
                         """)
         with gr.Column(scale=1):
+            # Video component remains for layout but won't show anything in this debug version
             video_output = gr.Video(label="Generated 3D Preview (DISABLED FOR DEBUG)", autoplay=False, loop=False, value=None, height=350)
             model_output = gr.Model3D(label="Extracted Model Preview", height=350, clear_color=[0.95, 0.95, 0.95, 1.0])
             with gr.Row():
@@ -391,8 +390,10 @@ with gr.Blocks(delete_cache=(600, 600), title="TRELLIS Text-to-3D") as demo:
 
     # --- Event Handlers ---
     print("Defining Gradio event handlers...")
-    demo.load(start_session) # Removed inputs/outputs kwargs, they are optional
-    demo.unload(end_session) # Removed incorrect inputs/outputs kwargs
+    # Use demo.load as in original user-provided code
+    demo.load(start_session, inputs=None, outputs=None)
+    # Use demo.unload as in original user-provided code (no extra args)
+    demo.unload(end_session) # Corrected: removed inputs/outputs
 
     generate_event = generate_btn.click(
         get_seed,
@@ -402,7 +403,7 @@ with gr.Blocks(delete_cache=(600, 600), title="TRELLIS Text-to-3D") as demo:
     ).then(
         text_to_3d,
         inputs=[text_prompt, seed, ss_guidance_strength, ss_sampling_steps, slat_guidance_strength, slat_sampling_steps],
-        outputs=[output_buf, video_output],
+        outputs=[output_buf, video_output], # state_dict -> output_buf, None -> video_output
         api_name="text_to_3d"
     ).then(
         lambda: (
@@ -458,10 +459,9 @@ if __name__ == "__main__":
     pipeline = None
     pipeline_loaded = False
     try:
-        # --- Load pipeline WITHOUT torch_dtype ---
+        # --- Load pipeline WITHOUT torch_dtype (As per original working version) ---
         pipeline = TrellisTextTo3DPipeline.from_pretrained(
             "JeffreyXiang/TRELLIS-text-xlarge"
-            # Removed: torch_dtype=torch.float16
         )
         if torch.cuda.is_available():
             pipeline = pipeline.to("cuda")
@@ -478,6 +478,7 @@ if __name__ == "__main__":
 
     if pipeline_loaded:
         print("Launching Gradio demo...")
+        # Use queue and debug=True as in original user-provided code
         demo.queue().launch(debug=True)
         print("Gradio demo launched.")
     else: