Update app.py

app.py

@@ -5,7 +5,9 @@ import requests
 import io
 from PIL import Image
 import os
-import matplotlib.image as mpimg
+import cv2
+import numpy as np
+
 # Create a text input
 
 api_key = os.getenv("ImageGenerating")
@@ -26,12 +28,16 @@ image_bytes = query({
 	"inputs": user_input,
 })
 
-def bytes_to_image_matplotlib(byte_data):
-    image = mpimg.imread(io.BytesIO(byte_data), format='PNG')
+def bytes_to_image_opencv(image_bytes):
+    np_arr = np.frombuffer(image_bytes, np.uint8)
+    image = cv2.imdecode(np_arr, cv2.IMREAD_COLOR)
     return image
-image = bytes_to_image_matplotlib(image_bytes)
+
+result_image_opencv = bytes_to_image_opencv(image_bytes)
+st.image(result_image_opencv,caption="image")
+
 # 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, caption=None, width=None, use_column_width=None, clamp=False, channels="RGB", output_format="PNG")
 # st.image(image=image,caption="image")
 # st.image(image)
 # st.image(image : image,caption=image)