Update app.py

app.py

@@ -85,7 +85,7 @@ print("Downloading LTX-2.3 distilled model + Gemma...")
 print("=" * 80)
 
 checkpoint_path = hf_hub_download(repo_id=LTX_MODEL_REPO, filename="ltx-2.3-22b-distilled.safetensors")
-spatial_upsampler_path = hf_hub_download(repo_id=LTX_MODEL_REPO, filename="ltx-2.3-spatial-upscaler-x2-1.1.safetensors")
+spatial_upsampler_path = hf_hub_download(repo_id=LTX_MODEL_REPO, filename="ltx-2.3-spatial-upscaler-x2-1.0.safetensors")
 gemma_root = snapshot_download(repo_id=GEMMA_REPO)
 
 print(f"Checkpoint: {checkpoint_path}")
@@ -102,7 +102,7 @@ pipeline = DistilledPipeline(
 )
 
 # Preload all models for ZeroGPU tensor packing.
-# DistilledPipeline does NOT have model_ledger — inspect at runtime instead.
+print("Preloading all models (including Gemma)...")
 print("Inspecting DistilledPipeline internals for ZeroGPU preloading...")
 _model_attrs = [a for a in dir(pipeline) if not a.startswith('_')]
 print(f"Available attributes: {_model_attrs}")
@@ -143,6 +143,10 @@ print("=" * 80)
 print("Pipeline ready!")
 print("=" * 80)
 
+print("=" * 80)
+print("Pipeline ready!")
+print("=" * 80)
+
 
 def log_memory(tag: str):
     if torch.cuda.is_available():
@@ -275,7 +279,7 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
                 lines=3,
                 placeholder="Describe the motion and animation you want...",
             )
-
+            
             with gr.Row():
                 duration = gr.Slider(label="Duration (seconds)", minimum=1.0, maximum=10.0, value=3.0, step=0.1)
                 with gr.Column():
@@ -324,4 +328,5 @@ css = """
 """
 
 if __name__ == "__main__":
-    demo.launch(theme=gr.themes.Citrus(), css=css)
+    demo.launch(theme=gr.themes.Citrus(), css=css)
+