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 |
def generate_image(prompt):
|
@@ -37,7 +37,7 @@ def generate_image(prompt):
|
|
37 |
|
38 |
|
39 |
text_gen = gr.Interface.load("models/Gustavosta/MagicPrompt-Stable-Diffusion")
|
40 |
-
proc1 = gr.Interface.load("models/
|
41 |
|
42 |
queue = Queue()
|
43 |
queue_threshold = 100
|
|
|
16 |
|
17 |
app = FastAPI()
|
18 |
|
19 |
+
API_URL = "https://api-inference.huggingface.co/models/ehristoforu/dalle-3-xl"
|
20 |
API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
|
21 |
headers = {"Authorization": f"Bearer {API_TOKEN}"}
|
22 |
def generate_image(prompt):
|
|
|
37 |
|
38 |
|
39 |
text_gen = gr.Interface.load("models/Gustavosta/MagicPrompt-Stable-Diffusion")
|
40 |
+
proc1 = gr.Interface.load("models/ehristoforu/dalle-3-xl")
|
41 |
|
42 |
queue = Queue()
|
43 |
queue_threshold = 100
|