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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -7,9 +7,8 @@ import os
7
  import requests
8
 
9
  # 환경변수에서 토큰 가져오기
10
- HF_API_TOKEN = os.getenv("HF_API_TOKEN")
11
- if not HF_API_TOKEN:
12
- raise ValueError("HF_API_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"
 
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"