Update app.py

app.py

@@ -5,7 +5,8 @@ import random
 import spaces
 from diffusers import DiffusionPipeline
 import torch
-
+import io
+import base64
 
 import numpy as np
 import random
@@ -106,9 +107,17 @@ def infer(
             height=height,
             generator=generator,
         ).images[0]
+
+
+    buffered = io.BytesIO()
+    image.save(buffered, format="WEBP")
+    img_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
+    img_data_url = f"data:image/webp;base64,{img_base64}"
+
+    print(img_data_url)
     
 
-    return image, seed
+    return image, img_data_url
 
 
 examples = [