24Sureshkumar commited on
Commit
39765db
·
verified ·
1 Parent(s): 639070c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -7,14 +7,13 @@ import re
7
  import os
8
  from huggingface_hub import login
9
 
10
- # Enter your Hugging Face token here (or set as environment variable)
11
- HF_TOKEN = "your_hf_token_here" # Replace with your actual token
 
12
 
13
  # Authenticate with Hugging Face Hub
14
- if HF_TOKEN:
15
- login(token=HF_TOKEN)
16
  else:
17
- raise ValueError("Hugging Face token not provided!")
18
 
19
  # Initialize models
20
  def load_models():
 
7
  import os
8
  from huggingface_hub import login
9
 
10
+ HF_TOKEN = os.environ.get("HF_TOKEN")
11
+ if not HF_TOKEN:
12
+ raise ValueError("Hugging Face token not found in environment variables!")
13
 
14
  # Authenticate with Hugging Face Hub
 
 
15
  else:
16
+ login(token=HF_TOKEN)
17
 
18
  # Initialize models
19
  def load_models():