Update app.py

app.py

@@ -1,12 +1,10 @@
 import gradio as gr
 import requests
 import io
-from PIL import Image
 import os
-from dotenv import load_dotenv
-load_dotenv()
+from PIL import Image
 
-# Define the API URL and headers
+# Get the API URL and headers from environment variables
 API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
 headers = {"Authorization": f"Bearer {os.getenv('HF_API_TOKEN')}"}
 
@@ -24,8 +22,8 @@ def generate_image(input_text):
 # Create Gradio interface
 iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.inputs.Textbox(label="Input Text", placeholder="Enter your description here..."),
-    outputs=gr.outputs.Image(type="pil", label="Generated Image"),
+    inputs=gr.Textbox(label="Input Text", placeholder="Enter your description here..."),
+    outputs=gr.Image(label="Generated Image"),
     title="Image Generation with FLUX",
     description="Enter a description to generate an image."
 )