Update app.py

app.py

@@ -2,6 +2,8 @@ import os
 import gradio as gr
 import requests
 import base64
+from io import BytesIO
+from PIL import Image  # Corrected import
 
 # Get API key from environment variable
 api_key = os.environ.get("NVCF_API_KEY")
@@ -36,7 +38,6 @@ def generate_image(prompt, negative_prompt, sampler, seed, guidance_scale, infer
         fetch_url = fetch_url_format + request_id
         response = session.get(fetch_url, headers=headers)
 
-
     response.raise_for_status()
     response_body = response.json()
 
@@ -50,12 +51,11 @@ def generate_image(prompt, negative_prompt, sampler, seed, guidance_scale, infer
     
     image_data = base64.b64decode(b64_image_data)
     
-    # Since Gradio's Image component can accept a PIL Image, convert the binary data to a PIL Image
+    # Convert the binary data to a PIL Image
     image = Image.open(BytesIO(image_data))
     
     return image
 
-
 # Create Gradio interface
 iface = gr.Interface(
     fn=generate_image,
@@ -71,4 +71,4 @@ iface = gr.Interface(
 )
 
 # Launch the Gradio app
-iface.launch()
+iface.launch()