Spaces:
Running
Running
chokiproai
commited on
Commit
•
b112a5a
1
Parent(s):
422e622
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ list_models = [
|
|
11 |
"Disney Pixar Cartoon", "Pixel Art XL", "Dalle 3 XL",
|
12 |
"Midjourney V4 XL", "Open Diffusion V1", "SSD 1B",
|
13 |
"Segmind Vega", "Animagine XL-2.0", "OpenDalle",
|
14 |
-
"OpenDalle V1.1", "
|
15 |
]
|
16 |
|
17 |
# Function to generate images from text
|
@@ -45,7 +45,7 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
|
|
45 |
API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/OpenDalle"
|
46 |
elif current_model == "OpenDalle V1.1":
|
47 |
API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/OpenDalleV1.1"
|
48 |
-
elif current_model == "
|
49 |
API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"
|
50 |
|
51 |
|
|
|
11 |
"Disney Pixar Cartoon", "Pixel Art XL", "Dalle 3 XL",
|
12 |
"Midjourney V4 XL", "Open Diffusion V1", "SSD 1B",
|
13 |
"Segmind Vega", "Animagine XL-2.0", "OpenDalle",
|
14 |
+
"OpenDalle V1.1", "PlaygroundV2 1024px aesthetic",
|
15 |
]
|
16 |
|
17 |
# Function to generate images from text
|
|
|
45 |
API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/OpenDalle"
|
46 |
elif current_model == "OpenDalle V1.1":
|
47 |
API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/OpenDalleV1.1"
|
48 |
+
elif current_model == "PlaygroundV2 1024px aesthetic":
|
49 |
API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"
|
50 |
|
51 |
|