Spaces:
Running
Running
chokiproai
commited on
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":
|