chokiproai commited on
Commit
9e33a06
1 Parent(s): b9e7f35

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -15,7 +15,8 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
15
  if current_model == "SD-1.5":
16
  API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
17
 
18
-
 
19
 
20
 
21
  if image_style == "None style":
 
15
  if current_model == "SD-1.5":
16
  API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
17
 
18
+ API_TOKEN = os.environ.get("HF_READ_TOKEN")
19
+ headers = {"Authorization": f"Bearer {API_TOKEN}"}
20
 
21
 
22
  if image_style == "None style":