Update app.py

app.py

@@ -9,7 +9,7 @@ from PIL import Image
 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",
+    "Midjourney-V4-XL", "Open-diffusion-v1", "SSD-1B",
 ]
 
 # Function to generate images from text
@@ -32,7 +32,9 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
     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"    
+        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"
 
     API_TOKEN = os.environ.get("HF_READ_TOKEN")
     headers = {"Authorization": f"Bearer {API_TOKEN}"}