gschurck commited on
Commit
a1ad261
·
1 Parent(s): 96d59f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def get_image_embedding(image):
15
  def get_text_embedding(text):
16
  multilingual_text_model = SentenceTransformer('clip-ViT-B-32-multilingual-v1')
17
  text_emb = multilingual_text_model.encode(text)
18
- return {"embedding": text_emb}
19
 
20
  image_embedding = gr.Interface(fn=get_image_embedding, inputs=gr.Image(type="pil"), outputs=gr.JSON())
21
  text_embedding = gr.Interface(fn=get_text_embedding, inputs=gr.Textbox(), outputs=gr.JSON())
 
15
  def get_text_embedding(text):
16
  multilingual_text_model = SentenceTransformer('clip-ViT-B-32-multilingual-v1')
17
  text_emb = multilingual_text_model.encode(text)
18
+ return {"embedding": text_emb.tolist()}
19
 
20
  image_embedding = gr.Interface(fn=get_image_embedding, inputs=gr.Image(type="pil"), outputs=gr.JSON())
21
  text_embedding = gr.Interface(fn=get_text_embedding, inputs=gr.Textbox(), outputs=gr.JSON())