fix

app.py

@@ -47,12 +47,7 @@ def infer(prompt, n_samples, steps, scale, seed):
     print(data)
     for image in data['output']['choices']:
         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)
+        images.append(im)
 
     # payload = {'prompt': prompt}
     # images_request = requests.post(url, json=payload)