Update video_service.py

video_service.py

@@ -29,7 +29,7 @@ def run_setup():
         print(f"ERRO CRÍTICO DURANTE O SETUP: 'setup.py' falhou com código {e.returncode}.")
         sys.exit(1)
 
-DEPS_DIR = Path("./deps")
+DEPS_DIR = Path("/data")
 LTX_VIDEO_REPO_DIR = DEPS_DIR / "LTX-Video"
 if not LTX_VIDEO_REPO_DIR.exists():
     run_setup()
@@ -111,9 +111,9 @@ class VideoService:
         models_dir = "downloaded_models_gradio"
         Path(models_dir).mkdir(parents=True, exist_ok=True)
         LTX_REPO = "Lightricks/LTX-Video"
-        distilled_model_path = hf_hub_download(repo_id=LTX_REPO, filename=self.config["checkpoint_path"], local_dir=models_dir, local_dir_use_symlinks=False)
+        distilled_model_path = hf_hub_download(repo_id=LTX_REPO, filename=self.config["checkpoint_path"], local_dir=models_dir, cache_dir=os.getenv("HF_HOME_CACHE"), token=os.getenv("HF_TOKEN"))
         self.config["checkpoint_path"] = distilled_model_path
-        spatial_upscaler_path = hf_hub_download(repo_id=LTX_REPO, filename=self.config["spatial_upscaler_model_path"], local_dir=models_dir, local_dir_use_symlinks=False)
+        spatial_upscaler_path = hf_hub_download(repo_id=LTX_REPO, filename=self.config["spatial_upscaler_model_path"], local_dir=models_dir, cache_dir=os.getenv("HF_HOME_CACHE"), token=os.getenv("HF_TOKEN"))
         self.config["spatial_upscaler_model_path"] = spatial_upscaler_path
         pipeline = create_ltx_video_pipeline(ckpt_path=self.config["checkpoint_path"], precision=self.config["precision"], text_encoder_model_name_or_path=self.config["text_encoder_model_name_or_path"], sampler=self.config["sampler"], device="cpu", enhance_prompt=False, prompt_enhancer_image_caption_model_name_or_path=self.config["prompt_enhancer_image_caption_model_name_or_path"], prompt_enhancer_llm_model_name_or_path=self.config["prompt_enhancer_llm_model_name_or_path"])
         latent_upsampler = None