ngoctuanai commited on
Commit
2f54470
·
1 Parent(s): 9836350

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,7 +5,8 @@ import random
5
  import os
6
  from PIL import Image
7
 
8
- API_URL = os.getenv("API_URL")
 
9
  API_TOKEN = os.getenv("HF_READ_TOKEN")
10
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
11
 
 
5
  import os
6
  from PIL import Image
7
 
8
+ API_URL = "https://api-inference.huggingface.co/models/CompVis/stable-diffusion-v1-4"
9
+ API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
10
  API_TOKEN = os.getenv("HF_READ_TOKEN")
11
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
12