Update app.py

app.py

@@ -8,13 +8,8 @@ API_URL = "https://router.huggingface.co/hf-inference/v1"
 headers = {"Authorization": "Bearer hf_xxxxxxxxxxxxxxxxxxxxxxxx"}
 
 def query(payload):
-	response = requests.post(API_URL, headers=headers, json=payload)
-	return response.content
-
-# You can access the image with PIL.Image for example
-import io
-from PIL import Image
-image = Image.open(io.BytesIO(image_bytes))
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.content
 
 # Load the FLUX.1-dev model from Hugging Face
 pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16)
@@ -22,6 +17,7 @@ pipe = pipe.to("cuda")
 
 # Define a function to generate an image based on the prompt
 def generate_image(prompt: str):
+    # Generate the image using the pipeline
     image = pipe(prompt).images[0]
     return image
 
@@ -34,3 +30,4 @@ iface = gr.Interface(fn=generate_image,
 
 # Launch the app
 iface.launch()
+