gschurck commited on
Commit
64a3664
·
1 Parent(s): 932317b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ def get_image_embedding(image):
8
  image_model = SentenceTransformer('clip-ViT-B-32')
9
 
10
  # Load and preprocess the image
11
- image = Image.open(BytesIO(image))
12
  img_emb = image_model.encode(image)
13
 
14
  return {"prediction": img_emb}
 
8
  image_model = SentenceTransformer('clip-ViT-B-32')
9
 
10
  # Load and preprocess the image
11
+ image = Image.open(image)
12
  img_emb = image_model.encode(image)
13
 
14
  return {"prediction": img_emb}