haepada commited on
Commit
638a6b2
·
verified ·
1 Parent(s): 932beb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -7,8 +7,9 @@ import os
7
  import requests
8
 
9
  # 환경변수에서 토큰 가져오기
10
- HF_API_TOKEN = os.getenv("HF_API_TOKEN", "HF_API_TOKEN")
11
- headers = {"Authorization": f"Bearer {HF_API_TOKEN}"}
 
12
 
13
  # Inference API 설정
14
  API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
 
7
  import requests
8
 
9
  # 환경변수에서 토큰 가져오기
10
+ HF_API_TOKEN = os.getenv("roots") # 변경된 부분
11
+ if not HF_API_TOKEN:
12
+ raise ValueError("roots token not found in environment variables")
13
 
14
  # Inference API 설정
15
  API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"