Update main.py

main.py

@@ -3,6 +3,7 @@ from pydantic import BaseModel
 from typing import List
 from PIL import Image
 import io
+import base64
 from sentence_transformers import SentenceTransformer, util
 
 class Item(BaseModel):
@@ -33,8 +34,8 @@ def predict(item: Item):
     #image_names = list(glob.glob('./*.jpg'))
     #print("Images:", len(image_names))
 
-    image_stream1 = io.BytesIO(item.image1);
-    image_stream2 = io.BytesIO(item.image2);
+    image_stream1 = io.BytesIO(base64.b64decode(item.image1));
+    image_stream2 = io.BytesIO(base64.b64decode(item.image2));
     
     encoded_image = model.encode([Image.open(image_stream1), Image.open(image_stream2)], batch_size=128, convert_to_tensor=True, show_progress_bar=True)