Update app.py

app.py

@@ -31,13 +31,11 @@ def generate_image(prompt, model_choice, translate=False):
     if translate:
         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 v1.5": "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5",
         "dalle-3-xl-v2": "https://api-inference.huggingface.co/models/ehristoforu/dalle-3-xl-v2",
         "midjourney-v6": "https://api-inference.huggingface.co/models/Kvikontent/midjourney-v6",
         "openjourney-v4": "https://api-inference.huggingface.co/models/prompthero/openjourney-v4",
         "LCM_Dreamshaper_v7": "https://api-inference.huggingface.co/models/SimianLuo/LCM_Dreamshaper_v7",
-        
-        
 
     }
     API_URL = model_urls[model_choice]
@@ -72,13 +70,13 @@ button {color: white; background-color: #106BA3; border: none; border-radius: 5p
 
 # Define interface
 title = "نموذج توليد الصور"
-description = "اكتب وصف للصورة التي تود من النظام التوليدي انشاءها"
+description = "اكتب وصف للصورة التي تود من النظام التوليدي انشاءها. على سبيل المثال: 'قطة ترتدي قبعة في مشهد شتوي'."
 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","dalle-3-xl-v2","midjourney-v6","openjourney-v4","LCM_Dreamshaper_v7"], label="Choose Model", value='Stable Diffusion v1.5'),
-        
+        gr.components.Checkbox(label="Translate The Text Before Generating Image", value=False)
     ],
     outputs=gr.components.Image(),
     title=title,
@@ -86,5 +84,5 @@ iface = gr.Interface(
     theme="default",
     css=css
 )
-
+# Launch the interface
 iface.launch()