uripper commited on
Commit
becee12
·
1 Parent(s): d78b29e
Files changed (1) hide show
  1. app/download_models.py +2 -0
app/download_models.py CHANGED
@@ -4,6 +4,8 @@ from huggingface_hub import hf_hub_download
4
  # Get the Hugging Face token from environment variable (automatically provided from the secret)
5
  HUGGINGFACE_TOKEN = os.getenv("key")
6
 
 
 
7
 
8
  def create_directory(dir_path):
9
  """
 
4
  # Get the Hugging Face token from environment variable (automatically provided from the secret)
5
  HUGGINGFACE_TOKEN = os.getenv("key")
6
 
7
+ if HUGGINGFACE_TOKEN is None:
8
+ raise ValueError("Hugging Face token is missing. Please set the 'key' environment variable.")
9
 
10
  def create_directory(dir_path):
11
  """