Update app.py

app.py

@@ -8,7 +8,8 @@ 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",
+    "Disney-Pixar-Cartoon", "Pixel-Art-XL", "Dalle-3-XL",
+    "Midjourney-V4-XL", "Playground-v2-1024px-aesthetic",
 ]
 
 # Function to generate images from text
@@ -29,7 +30,9 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
     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"    
+        API_URL = "https://api-inference.huggingface.co/models/openskyml/midjourney-v4-xl"
+    elif current_model == "Playground-v2-1024px-aesthetic":
+        API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"    
 
     API_TOKEN = os.environ.get("HF_READ_TOKEN")
     headers = {"Authorization": f"Bearer {API_TOKEN}"}