Update app.py

app.py

@@ -111,11 +111,10 @@ def infer(
     
     for prediction in json_data["predictions"]:
         for image in prediction["images"]:
-            image_b64 = (f"data:image/jpeg;base64,{image}")
-            images.append(image_b64)
+            pil_image = Image.open(BytesIO(base64.b64decode(image)))
+            images.append(pil_image)
 
             if profile is not None: # avoid conversion on non-logged-in users
-                pil_image = Image.open(BytesIO(base64.b64decode(image)))
                 user_history.save_image( # save images + metadata to user history
                     label=prompt,
                     image=pil_image,