Update app.py

app.py

@@ -5,6 +5,7 @@ import torchvision
 from diffusers import DiffusionPipeline
 import PIL.Image
 import numpy as np
+from io import BytesIO
 
 ldm = DiffusionPipeline.from_pretrained("fusing/latent-diffusion-text2im-large")
 
@@ -19,10 +20,11 @@ def greet(name):
     image_processed = image_processed.numpy().astype(np.uint8)
     image_pil = PIL.Image.fromarray(image_processed[0])
     
-    # save image
-    image_pil.save("test.png")
-    encoded_string= base64.b64encode(image_pil.read())
-    print(encoded_string.decode('utf-8'))
+    # save image as buffer
+    buffered = BytesIO()
+    image_pil.save(buffered, format="JPEG")
+    img_str = base64.b64encode(buffered.getvalue())
+    print(img_str.decode('utf-8'))
     return image_pil
     #return "Gello " + prompt + "!!"