Update app.py

app.py

@@ -9,7 +9,7 @@ from pathlib import Path
 import spaces
 import gradio as gr
 import torch
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download,snapshot_download
 
 # Import from public LTX-2 package
 # Install with: pip install git+https://github.com/Lightricks/LTX-2.git
@@ -21,6 +21,7 @@ DEFAULT_REPO_ID = "Lightricks/LTX-2"
 DEFAULT_GEMMA_REPO_ID = "google/gemma-3-12b-it-qat-q4_0-unquantized"
 DEFAULT_CHECKPOINT_FILENAME = "ltx-2-19b-dev-fp8.safetensors"
 
+
 def get_hub_or_local_checkpoint(repo_id: str, filename: str):
     """Download from HuggingFace Hub."""
     print(f"Downloading {filename} from {repo_id}...")
@@ -28,19 +29,28 @@ def get_hub_or_local_checkpoint(repo_id: str, filename: str):
     print(f"Downloaded to {ckpt_path}")
     return ckpt_path
 
+def download_gemma_model(repo_id: str):
+    """Download the full Gemma model directory."""
+    print(f"Downloading Gemma model from {repo_id}...")
+    local_dir = snapshot_download(repo_id=repo_id)
+    print(f"Gemma model downloaded to {local_dir}")
+    return local_dir
+
 # Initialize model ledger and text encoder at startup (load once, keep in memory)
 print("=" * 80)
 print("Loading Gemma Text Encoder...")
 print("=" * 80)
 
 checkpoint_path = get_hub_or_local_checkpoint(DEFAULT_REPO_ID, DEFAULT_CHECKPOINT_FILENAME)
+gemma_local_path = download_gemma_model(DEFAULT_GEMMA_REPO_ID)
 device = "cuda"
 
 print(f"Initializing text encoder with:")
 print(f"  checkpoint_path={checkpoint_path}")
-print(f"  gemma_root={DEFAULT_GEMMA_REPO_ID}")
+print(f"  gemma_root={gemma_local_path}")
 print(f"  device={device}")
 
+
 model_ledger = ModelLedger(
     dtype=torch.bfloat16,
     device=device,