Update app.py

app.py

@@ -1,5 +1,8 @@
 import gradio as gr
 import requests
+from PIL import Image
+from io import BytesIO
+import base64
 
 api_url = "https://5cb20b40-572c-426f-9466-995256f9b6eb.id.repl.co/generate_image"
 
@@ -7,7 +10,10 @@ def generate_image(prompt, seed=0, negative_prompt="", model="Deliberate", sampl
     data = "?prompt="+ prompt + "&seed="+ str(seed) + "&negative_prompt=" + negative_prompt + "&model=" + model + "&sampler=" + sampler + "&steps=" + str(steps)
     response = requests.post(api_url, data=data)
     if response.status_code == 200:
-        return response.json()["url"]
+        img_base64 = response.json()["url"]
+        img_bytes = base64.b64decode(img_base64)
+        img = Image.open(BytesIO(img_bytes))
+        return img
     else:
         return None
 
@@ -28,4 +34,4 @@ interface = gr.Interface(generate_image, inputs, outputs, title="<a href='https:
                                    ["A landscape with mountains and a lake"], 
                                    ["A portrait of a woman with red hair"]])
 
-interface.launch()
+interface.launch()