Update app.py

app.py

@@ -27,9 +27,9 @@ def query(payload, API_URL, headers):
     return response.content
 
 # Function to generate images based on prompts using the Hugging Face API
-def generate_image(prompt, model_choice):
-    
-    prompt = translate_text(prompt, target_language='en')  # Assuming you want to translate to English
+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",
         "dalle-3-xl-v2": "https://api-inference.huggingface.co/models/ehristoforu/dalle-3-xl-v2",
@@ -78,7 +78,7 @@ iface = gr.Interface(
     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,