Update main.py
Browse files
main.py
CHANGED
@@ -16,7 +16,7 @@ import base64
|
|
16 |
|
17 |
app = FastAPI()
|
18 |
|
19 |
-
API_URL = "https://api-inference.huggingface.co/models/
|
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/
|
29 |
|
30 |
queue = Queue()
|
31 |
queue_threshold = 100
|
|
|
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 |
|
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
|