Uppdate

.gitignore

@@ -180,4 +180,5 @@ cython_debug/
 /datasets
 /wandb
 
-samples/
+samples/
+.gradio

app.py

@@ -44,7 +44,13 @@ def load_model_components(device: str = "cuda"):
     try:
         # Load configuration
         config_path = "configs/t2i/tim_xl_p1_t2i.yaml"
-        ckpt_path = "checkpoints/t2i_model.bin"
+        from huggingface_hub import hf_hub_download
+
+        ckpt_path = hf_hub_download(
+            repo_id="blanchon/TiM-checkpoints",
+            filename="t2i_model.bin",
+            local_dir=Path(__file__).parent,
+        )
 
         if not Path(config_path).exists():
             raise FileNotFoundError(f"Config file not found: {config_path}")