Update app.py

app.py

@@ -1,4 +1,5 @@
 import glob
+import os
 from copy import deepcopy
 
 import gradio as gr
@@ -15,6 +16,8 @@ from torchvision.transforms import ToPILImage, ToTensor
 from transformers import AutoModelForImageSegmentation
 from utils import extract_object, get_model_from_config, resize_and_center_crop
 
+huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
+
 ASPECT_RATIOS = {
     str(512 / 2048): (512, 2048),
     str(1024 / 1024): (1024, 1024),
@@ -30,8 +33,8 @@ ASPECT_RATIOS = {
 }
 
 # download the config and model
-MODEL_PATH = hf_hub_download("jasperai/LBM_relighting", "relight.safetensors")
-CONFIG_PATH = hf_hub_download("jasperai/LBM_relighting", "relight.yaml")
+MODEL_PATH = hf_hub_download("jasperai/LBM_relighting", "relight.safetensors", token=huggingface_token)
+CONFIG_PATH = hf_hub_download("jasperai/LBM_relighting", "relight.yaml", token=huggingface_token)
 
 with open(CONFIG_PATH, "r") as f:
     config = yaml.safe_load(f)