Update app.py

app.py

@@ -49,9 +49,10 @@ def query(payload):
 def generate_image_from_prompt(prompt_text):
     gr.Info("Image generation started")
     image_bytes = query({"inputs": prompt_text})
-    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
+    img = BytesIO(image_bytes)
+    pil_img = Image.open(img)
+    gr.Info("Image Generated Successfully!")
+    return pil_img
 
 title = "3D Icons SDXL Demo 🎨"
 description = "This app uses Hugging Face AI model to generate an image based on the provided text prompt 🖼."