gschurck commited on
Commit
fb76362
·
1 Parent(s): c0b0792

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -10,7 +10,8 @@ def get_image_embedding(image):
10
  # Load and preprocess the image
11
  img_emb = image_model.encode(image)
12
  print(img_emb)
13
- return {"prediction": img_emb}
 
14
 
15
 
16
  image_input = gr.Image(type="pil")
 
10
  # Load and preprocess the image
11
  img_emb = image_model.encode(image)
12
  print(img_emb)
13
+ print(type(img_emb))
14
+ return {"prediction": img_emb.tolist()}
15
 
16
 
17
  image_input = gr.Image(type="pil")