varshakolanu commited on
Commit
3fcf773
·
verified ·
1 Parent(s): c435100

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -11,7 +11,10 @@ import os
11
 
12
  app = FastAPI()
13
 
14
- API_KEY = "hf_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" # Replace with your Hugging Face API key
 
 
 
15
 
16
  class VendorLog(BaseModel):
17
  vendorLogId: str
 
11
 
12
  app = FastAPI()
13
 
14
+ # Load API key from environment variable
15
+ API_KEY = os.getenv("HUGGINGFACE_API_KEY")
16
+ if not API_KEY:
17
+ raise ValueError("HUGGINGFACE_API_KEY environment variable not set")
18
 
19
  class VendorLog(BaseModel):
20
  vendorLogId: str