Update app.py

app.py

@@ -10,7 +10,7 @@ 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",
+    "Segmind Vega", "Animagine XL-2.0", "FunkoPOP",
 ]
 
 # Function to generate images from text
@@ -39,7 +39,9 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
     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"    
+        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}"}