Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,9 +11,11 @@ def get_image_embedding(image):
|
|
| 11 |
def get_text_embedding(text):
|
| 12 |
multilingual_text_model = SentenceTransformer('clip-ViT-B-32-multilingual-v1')
|
| 13 |
text_emb = multilingual_text_model.encode(text)
|
|
|
|
|
|
|
| 14 |
return {"embedding": text_emb.tolist()}
|
| 15 |
|
| 16 |
-
image_embedding = gr.Interface(fn=get_image_embedding, inputs=gr.Image(type="
|
| 17 |
text_embedding = gr.Interface(fn=get_text_embedding, inputs=gr.Textbox(), outputs=gr.JSON(api_name="text-embedding"), title="Text Embedding")
|
| 18 |
|
| 19 |
space = gr.TabbedInterface([image_embedding, text_embedding], ["Image Embedding", "Text Embedding"])
|
|
|
|
| 11 |
def get_text_embedding(text):
|
| 12 |
multilingual_text_model = SentenceTransformer('clip-ViT-B-32-multilingual-v1')
|
| 13 |
text_emb = multilingual_text_model.encode(text)
|
| 14 |
+
print(text_emb)
|
| 15 |
+
print(text_emb.shape)
|
| 16 |
return {"embedding": text_emb.tolist()}
|
| 17 |
|
| 18 |
+
image_embedding = gr.Interface(fn=get_image_embedding, inputs=gr.Image(type="filepath"), outputs=gr.JSON(api_name="image-embedding"), title="Image Embedding")
|
| 19 |
text_embedding = gr.Interface(fn=get_text_embedding, inputs=gr.Textbox(), outputs=gr.JSON(api_name="text-embedding"), title="Text Embedding")
|
| 20 |
|
| 21 |
space = gr.TabbedInterface([image_embedding, text_embedding], ["Image Embedding", "Text Embedding"])
|