Update app.py

app.py

@@ -1,22 +1,17 @@
-## app.py:
 import torch
 import gradio as gr
-from diffusers import StableDiffusionPipeline
 import requests
 from io import BytesIO
-import os
 from PIL import Image
-
-
+import os
 
 def translate_text(text, target_language='en'):
     API_URL = "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-ar-en"
     headers = {"Authorization": f"Bearer {os.getenv('API_TOKEN')}"}
-    response = requests.post(API_URL, headers=headers, json=text)
+    response = requests.post(API_URL, headers=headers, json={"inputs": text})
 
     if response.status_code == 200:
         return response.json()[0]['translation_text']
-
     else:
         print("Failed to translate text:", response.text)
         return text  # Return the original text if translation fails
@@ -30,14 +25,13 @@ def query(payload, API_URL, headers):
 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",
         "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]
 
@@ -49,10 +43,6 @@ def generate_image(prompt, model_choice, translate=False):
         image = Image.open(BytesIO(data))
         # Resize the image
         image = image.resize((400, 400))
-        # Convert the image object back to bytes for Gradio output
-        buf = BytesIO()
-        image.save(buf, format='PNG')
-        buf.seek(0)
         return image
 
     except Exception as e:
@@ -75,11 +65,11 @@ 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)
+        gr.Textbox(lines=2, placeholder="Enter the description of the image here..."),
+        gr.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.Checkbox(label="Translate The Text Before Generating Image", value=False)
     ],
-    outputs=gr.components.Image(),
+    outputs=gr.Image(),
     title=title,
     description=description,
     theme="default",