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/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
|