Update app.py

app.py

@@ -32,6 +32,11 @@ def generate_image(prompt, model_choice, translate=False):
         prompt = translate_text(prompt, target_language='en')  # Assuming you want to translate to English
     model_urls = {
         "Stable Diffusion v1.5": "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5",
+        "stable-diffusion-xl-base-1.0": "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0",
+        "sdxl-turbo": "https://api-inference.huggingface.co/models/stabilityai/sdxl-turbo",
+        "MistoLine": "https://api-inference.huggingface.co/models/TheMistoAI/MistoLine",
+        "Hyper-SD": "https://api-inference.huggingface.co/models/ByteDance/Hyper-SD",
+        
 
     }
     API_URL = model_urls[model_choice]
@@ -71,7 +76,7 @@ iface = gr.Interface(
     fn=generate_image,
     inputs=[
         gr.components.Textbox(lines=2, placeholder="Enter the description of the image here..."),
-        gr.components.Dropdown(choices=["Stable Diffusion v1.5",], label="Choose Model", value='Stable Diffusion v1.5'),
+        gr.components.Dropdown(choices=["Stable Diffusion v1.5","stable-diffusion-xl-base-1.0","sdxl-turbo","MistoLine","Hyper-SD"], label="Choose Model", value='Stable Diffusion v1.5'),
         gr.components.Checkbox(label="Translate The Text Before Generating Image", value=False)
     ],
     outputs=gr.components.Image(),