fix base64

app.py

@@ -10,6 +10,7 @@ from io import BytesIO
 import os
 import requests
 import json
+import base64
 # from urllib import parse
 
 from share_btn import community_icon_html, loading_icon_html, share_js
@@ -45,7 +46,12 @@ def infer(prompt, n_samples, steps, scale, seed):
     data = json.load(BytesIO(response.content))
     print(data)
     for image in data['output']['choices']:
-        image_b64 = (f"data:image/jpeg;base64,{image['image_base64']}")
+        im = Image.open(BytesIO(base64.b64decode(image['image_base64'])))
+        im_file = BytesIO()
+        im.save(im_file, format="JPEG")
+        im_bytes = im_file.getvalue()  # im_bytes: image in binary format.
+        im_b64 = base64.b64encode(im_bytes)
+        image_b64 = (f"data:image/jpeg;base64,{im_b64}")
         images.append(image_b64)
 
     # payload = {'prompt': prompt}