Update handler.py

handler.py

@@ -25,6 +25,10 @@ from varnish import Varnish
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+
+# Get token from environment
+hf_token = os.getenv("HF_API_TOKEN")
+
 # Constraints
 MAX_LARGE_SIDE = 1280
 MAX_SMALL_SIDE = 768 # should be 720 but it must be divisible by 32
@@ -524,12 +528,14 @@ class EndpointHandler:
                     if hasattr(self.text_to_video, 'load_lora_weights'):
                         self.text_to_video.load_lora_weights(
                             config.lora_model_name,
-                            weight_name=config.lora_model_weight_file if config.lora_model_weight_file else None
+                            weight_name=config.lora_model_weight_file if config.lora_model_weight_file else None,
+                            token=hf_token,
                         )
                     if hasattr(self.image_to_video, 'load_lora_weights'):
                         self.image_to_video.load_lora_weights(
                             config.lora_model_name,
-                            weight_name=config.lora_model_weight_file if config.lora_model_weight_file else None
+                            weight_name=config.lora_model_weight_file if config.lora_model_weight_file else None,
+                            token=hf_token,
                         )
                     self._current_lora_model = (config.lora_model_name, config.lora_model_weight_file)