Update app.py

app.py

@@ -1,12 +1,20 @@
+import os
 import cv2
 import numpy as np
 import onnxruntime as ort
 import gradio as gr
 from huggingface_hub import hf_hub_download
 
-# Download the model from Hugging Face Hub
+# Ensure the Hugging Face token is retrieved from environment variables
+huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
+
+# Check if the token is available
+if huggingface_token is None:
+    raise ValueError("HUGGINGFACE_TOKEN environment variable not set.")
+
+# Download the model file from Hugging Face using the token
 model_id = "Arrcttacsrks/netrunner-exe_Insight-Swap-models-onnx"
-model_file = hf_hub_download(repo_id=model_id, filename="simswap_512_unoff.onnx")
+model_file = hf_hub_download(repo_id=model_id, filename="simswap_512_unoff.onnx", token=huggingface_token)
 
 def load_and_preprocess_image(image):
     img = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)