Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from PIL import Image
 import requests
 import io
+import base64
 
 # Set page config
 st.set_page_config(page_title="AI Image Generator", page_icon="🎨", layout="wide")
@@ -36,7 +37,7 @@ if st.button("Generate Image"):
                 
                 if response.status_code == 200:
                     image_data = response.json()['images'][0]
-                    image = Image.open(io.BytesIO(Image.open(io.BytesIO(image_data)).tobytes()))
+                    image = Image.open(io.BytesIO(base64.b64decode(image_data)))
                     
                     # Display the image
                     st.image(image, caption="Generated Image", use_column_width=True)