Spaces:
Running
Running
chokiproai
commited on
Commit
•
5ce2a2e
1
Parent(s):
b6711ff
UPDATE MODELS
Browse files
app.py
CHANGED
@@ -10,8 +10,8 @@ list_models = [
|
|
10 |
"SDXL 1.0", "SD 1.5", "OpenJourney", "Anything V4.0",
|
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", "
|
14 |
-
"OpenDalle V1.1", "PlaygroundV2 1024px aesthetic",
|
15 |
]
|
16 |
|
17 |
# Function to generate images from text
|
@@ -41,6 +41,8 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
|
|
41 |
API_URL = "https://api-inference.huggingface.co/models/segmind/Segmind-Vega"
|
42 |
elif current_model == "Animagine XL-2.0":
|
43 |
API_URL = "https://api-inference.huggingface.co/models/Linaqruf/animagine-xl-2.0"
|
|
|
|
|
44 |
elif current_model == "OpenDalle":
|
45 |
API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/OpenDalle"
|
46 |
elif current_model == "OpenDalle V1.1":
|
|
|
10 |
"SDXL 1.0", "SD 1.5", "OpenJourney", "Anything V4.0",
|
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", "Animagine XL-3.0",
|
14 |
+
"OpenDalle", "OpenDalle V1.1", "PlaygroundV2 1024px aesthetic",
|
15 |
]
|
16 |
|
17 |
# Function to generate images from text
|
|
|
41 |
API_URL = "https://api-inference.huggingface.co/models/segmind/Segmind-Vega"
|
42 |
elif current_model == "Animagine XL-2.0":
|
43 |
API_URL = "https://api-inference.huggingface.co/models/Linaqruf/animagine-xl-2.0"
|
44 |
+
elif current_model == "Animagine XL-3.0":
|
45 |
+
API_URL = "https://api-inference.huggingface.co/models/cagliostrolab/animagine-xl-3.0"
|
46 |
elif current_model == "OpenDalle":
|
47 |
API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/OpenDalle"
|
48 |
elif current_model == "OpenDalle V1.1":
|