Update app.py

app.py

@@ -2,29 +2,24 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 
-# Load API key from environment variable
 api_key = os.getenv("HF_API_KEY")
-
-# Initialize the InferenceClient
 client = InferenceClient(provider="auto", api_key=api_key)
-
-# List of available models
 model_list = ["Qwen/Qwen-Image", "black-forest-labs/FLUX.1-dev"]
 
-# Function to generate image from text prompt
 def generate_image(prompt, model_name):
     image = client.text_to_image(prompt, model=model_name)
     return image
 
-# Create Gradio interface
-with gr.Blocks() as demo:
-    gr.Markdown("## Text to Image Generator using Hugging Face Inference API")
-    with gr.Row():
-        prompt_input = gr.Textbox(label="Enter your prompt", placeholder="e.g. a girl playing on ground")
-        model_dropdown = gr.Dropdown(choices=model_list, label="Select Model", value=model_list[0])
-    generate_button = gr.Button("Generate Image")
-    output_image = gr.Image(type="pil", label="Generated Image")
+def app():
+    with gr.Blocks() as demo:
+        gr.Markdown("## Text to Image Generator")
+        prompt = gr.Textbox(label="Prompt")
+        model = gr.Dropdown(choices=model_list, label="Model", value=model_list[0])
+        btn = gr.Button("Generate")
+        output = gr.Image(type="pil")
 
-    generate_button.click(fn=generate_image, inputs=[prompt_input, model_dropdown], outputs=output_image)
+        btn.click(generate_image, inputs=[prompt, model], outputs=output)
+    return demo
 
+demo = app()
 demo.launch()