Update app.py

app.py

@@ -6,20 +6,27 @@ 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",
+]
 
 # 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 == "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 == "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":
@@ -42,7 +49,7 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
         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_TOKEN = os.environ.get("HF_READ_TOKEN")
     headers = {"Authorization": f"Bearer {API_TOKEN}"}