Update app.py

app.py

@@ -47,8 +47,9 @@ def query(payload):
 
 def generate_image_from_prompt(prompt_text):
     image_bytes = query({"inputs": prompt_text})
-    generated_image = Image.open(io.BytesIO(image_bytes))
-    return generated_image
+    img = BytesIO(image_bytes)   # Convert to BytesIO stream
+    pil_img = Image.open(img)    # Open the image using PIL library
+    return pil_img               # Return the converted PIL image
 
 title = "KVIImager 2.0 Demo 🎨"
 description = "This app uses Hugging Face AI model to generate an image based on the provided text prompt 🖼️."
@@ -56,13 +57,13 @@ description = "This app uses Hugging Face AI model to generate an image based on
 input_prompt = gr.Textbox(label="Enter Prompt 📝", placeholder="E.g. 'A peaceful garden with a small cottage'")
 output_generated_image = gr.Image(label="Generated Image")
 
-with gr.Blocks(theme=gr.themes.Soft()) as app:
-    caption = "Generate Image"
-    iface = gr.Interface(
-        generate_image_from_prompt,
-        inputs=input_prompt, 
-        outputs=output_generated_image, 
-        title=title,
-        description=description
-    )
-    iface.launch()
+#with gr.Blocks(theme=gr.themes.Soft()) as app:
+   # caption = "Generate Image"
+iface = gr.Interface(
+    generate_image_from_prompt,
+    inputs=input_prompt, 
+    outputs=output_generated_image, 
+    title=title,
+    description=description
+)
+iface.launch()