Ashrafb commited on
Commit
aa875bc
1 Parent(s): ea3760e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -16,7 +16,7 @@ import base64
16
 
17
  app = FastAPI()
18
 
19
- API_URL = "https://api-inference.huggingface.co/models/playground-v2-1024px-aesthetic"
20
  API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
21
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
22
 
@@ -25,7 +25,7 @@ headers = {"Authorization": f"Bearer {API_TOKEN}"}
25
 
26
 
27
  text_gen = gr.Interface.load("models/Gustavosta/MagicPrompt-Stable-Diffusion")
28
- proc1 = gr.Interface.load("models/playground-v2-1024px-aesthetic")
29
 
30
  queue = Queue()
31
  queue_threshold = 100
 
16
 
17
  app = FastAPI()
18
 
19
+ API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"
20
  API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
21
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
22
 
 
25
 
26
 
27
  text_gen = gr.Interface.load("models/Gustavosta/MagicPrompt-Stable-Diffusion")
28
+ proc1 = gr.Interface.load("models/playgroundai/playground-v2-1024px-aesthetic")
29
 
30
  queue = Queue()
31
  queue_threshold = 100