Update app.py
Browse files
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(
|
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}
|