Update app.py

app.py

@@ -6,14 +6,6 @@ import os
 from PIL import Image
 
 # List of available models
-"""
-list_models = [
-    "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", "FunkoPOP",
-]
-"""
 
 list_models = [
     "FunkoPOP", "SD 1.5",
@@ -25,31 +17,7 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
     if current_model == "FunkoPOP":
         API_URL = "https://api-inference.huggingface.co/models/ProomptEngineer/pe-funko-pop-diffusion-style"
     elif current_model == "SD 1.5":
-        API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
-"""
-    elif current_model == "OpenJourney":
-        API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney"       
-    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":
-        API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/disney-pixar-cartoon"
-    elif current_model == "Pixel Art XL":
-        API_URL = "https://api-inference.huggingface.co/models/nerijs/pixel-art-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":
-        API_URL = "https://api-inference.huggingface.co/models/openskyml/midjourney-v4-xl"
-    elif current_model == "Open Diffusion V1":
-        API_URL = "https://api-inference.huggingface.co/models/openskyml/open-diffusion-v1" 
-    elif current_model == "SSD 1B":
-        API_URL = "https://api-inference.huggingface.co/models/segmind/SSD-1B"
-    elif current_model == "Segmind Vega":
-        API_URL = "https://api-inference.huggingface.co/models/segmind/Segmind-Vega"
-    elif current_model == "Animagine XL-2.0":
-        API_URL = "https://api-inference.huggingface.co/models/Linaqruf/animagine-xl-2.0"   
-    elif current_model == "FunkoPOP":
-        API_URL = "https://api-inference.huggingface.co/models/ProomptEngineer/pe-funko-pop-diffusion-style"   
-"""
+        API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"   
     API_TOKEN = os.environ.get("HF_READ_TOKEN")
     headers = {"Authorization": f"Bearer {API_TOKEN}"}