chokiproai commited on
Commit
b112a5a
1 Parent(s): 422e622

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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", "Playground v2 1024px aesthetic",
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 == "Playground v2 1024px aesthetic":
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