Update app.py

app.py

@@ -7,38 +7,38 @@ from PIL import Image
 
 # List of available models
 list_models = [
-    "SDXL-1.0", "SD-1.5", "OpenJourney-V4", "Anything-V4",
-    "Disney-Pixar-Cartoon", "Pixel-Art-XL", "Dalle-3-XL",
-    "Midjourney-V4-XL", "Open-diffusion-v1", "SSD-1B",
-    "Segmind-Vega", "Animagine-XL-2.0",
+    "SDXL 1.0", "SD 1.5", "OpenJourney", "Anything V4.0",
+    "Disney Pixar Cartoon", "Pixel Art XL", "Dalle 3 XL",
+    "Midjourney V4 XL", "Open Diffusion V1", "SSD 1B",
+    "Segmind Vega", "Animagine XL-2.0",
 ]
 
 # Function to generate images from text
 def generate_txt2img(current_model, prompt, is_negative=False, image_style="None style", steps=50, cfg_scale=7, seed=None):
 
-    if current_model == "SD-1.5":
+    if current_model == "SD 1.5":
         API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
-    elif current_model == "SDXL-1.0":
+    elif current_model == "SDXL 1.0":
         API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
     elif current_model == "OpenJourney":
         API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney"       
-    elif current_model == "Anything-V4.0":
+    elif current_model == "Anything V4.0":
         API_URL = "https://api-inference.huggingface.co/models/xyn-ai/anything-v4.0" 
-    elif current_model == "Disney-Pixar-Cartoon":
+    elif current_model == "Disney Pixar Cartoon":
         API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/disney-pixar-cartoon"
-    elif current_model == "Pixel-Art-XL":
+    elif current_model == "Pixel Art XL":
         API_URL = "https://api-inference.huggingface.co/models/nerijs/pixel-art-xl"
-    elif current_model == "Dalle-3-XL":
+    elif current_model == "Dalle 3 XL":
         API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
-    elif current_model == "Midjourney-V4-XL":
+    elif current_model == "Midjourney V4 XL":
         API_URL = "https://api-inference.huggingface.co/models/openskyml/midjourney-v4-xl"
-    elif current_model == "Open-Diffusion-V1":
+    elif current_model == "Open Diffusion V1":
         API_URL = "https://api-inference.huggingface.co/models/openskyml/open-diffusion-v1" 
-    elif current_model == "SSD-1B":
+    elif current_model == "SSD 1B":
         API_URL = "https://api-inference.huggingface.co/models/segmind/SSD-1B"
-    elif current_model == "Segmind-Vega":
+    elif current_model == "Segmind Vega":
         API_URL = "https://api-inference.huggingface.co/models/segmind/Segmind-Vega"
-    elif current_model == "Animagine-XL-2.0":
+    elif current_model == "Animagine XL-2.0":
         API_URL = "https://api-inference.huggingface.co/models/Linaqruf/animagine-xl-2.0"    
 
     API_TOKEN = os.environ.get("HF_READ_TOKEN")