Update app.py

app.py

@@ -4,17 +4,15 @@ from PIL import Image
 import io
 import os
 
-# Load the Hugging Face API key from environment variables
-API_KEY = os.getenv("HF_API_TOKEN")  # Replace with environment variable reference
+API_KEY = os.getenv("HF_API_TOKEN")
 MODEL_NAME = "black-forest-labs/FLUX.1-dev"
 
-# Initialize client
 client = InferenceClient(api_key=API_KEY)
 
 def generate_image(prompt):
     """Generate an image from a text prompt using Hugging Face API."""
     try:
-        image = client.text_to_image(prompt, model=MODEL_NAME)  # Returns a PIL image
+        image = client.text_to_image(prompt, model=MODEL_NAME)
         img_byte_array = io.BytesIO()
         image.save(img_byte_array, format="PNG")
         img_byte_array.seek(0)
@@ -65,7 +63,7 @@ input, textarea {
 with gr.Blocks(css=css) as demo:
     title="🚀 Agents Valley AI Image Generator",
     gr.Markdown(
-        "<h2 style='color: white; font-size: 18px; font-weight: 500;'>Enter a description and generate an AI image powered by Agents Valley!</h2>"
+        "<h2 style='color: white; font-size: 18px; font-weight: 500;'>AI image Generator powered by Agents Valley</h2>"
     )
     prompt_input = gr.Textbox(label="Enter your prompt", value="A robot on a stallion.")
     output_image = gr.Image(label="Generated Image by Agents Valley")