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