Changes

app.py

@@ -54,8 +54,11 @@ def generate():
 
         image_bytes = query({"inputs": prompt,})
         # You can access the image with PIL.Image for example
+        img_io = io.BytesIO(image_bytes)
 
-        image = Image.open(io.BytesIO(image_bytes))
+        image = Image.open(img_io)
+        img_io.seek(0)
+        byteImg = img_io.read()
 
         # guidance_scale = 7.5
         # num_inference_steps = 24
@@ -79,7 +82,7 @@ def generate():
         #         generator=g_cuda
         #     ).images
         
-        st.image(image)
+        st.image(byteImg)
     
     else:
         st.write('<Enter parameters to generate image>')