chokiproai commited on
Commit
5ce2a2e
1 Parent(s): b6711ff

UPDATE MODELS

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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", "OpenDalle",
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":