Update app.py

app.py

@@ -10,6 +10,7 @@ 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",
 ]
 
 # Function to generate images from text
@@ -35,6 +36,10 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
         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"    
 
     API_TOKEN = os.environ.get("HF_READ_TOKEN")
     headers = {"Authorization": f"Bearer {API_TOKEN}"}