Update app.py

app.py

@@ -5,6 +5,7 @@ import requests
 import io
 from PIL import Image
 import os
+import matplotlib.image as mpimg
 # Create a text input
 
 api_key = os.getenv("ImageGenerating")
@@ -25,7 +26,11 @@ image_bytes = query({
 	"inputs": user_input,
 })
 
-image = Image.open(io.BytesIO(image_bytes))
+def bytes_to_image_matplotlib(byte_data):
+    image = mpimg.imread(io.BytesIO(byte_data), format='PNG')
+    return image
+image = bytes_to_image_matplotlib(image_bytes)
+# image = Image.open(io.BytesIO(image_bytes))
 st.image(image, caption=None, width=None, use_column_width=None, clamp=False, channels="RGB", output_format="PNG")
 # st.image(image=image,caption="image")
 # st.image(image)