Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import requests
 import os
 import base64
+from PIL import Image
 
 
 api_url = "https://codinglehrer.pythonanywhere.com/generate_image"
@@ -20,15 +21,11 @@ def generate_image(prompt, seed=0, negative_prompt="", model="Deliberate", sampl
         # Decode the base64 image string
         img_bytes = base64.b64decode(response.json()["url"])
         
-        # Save the image in the images folder
-        if not os.path.exists("images"):
-            os.makedirs("images")
-        filename = f"images/{seed}.png"
-        with open(filename, 'wb') as f:
-            f.write(img_bytes)
+        # Open the image using Pillow
+        img = Image.open(io.BytesIO(img_bytes))
         
-        # Return the path of the saved image file
-        return filename
+        # Return the image object
+        return img
     else:
         return None