Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def get_text_embedding(text):
|
|
16 |
print(text_emb.ndim)
|
17 |
return {"embedding": text_emb.tolist()}
|
18 |
|
19 |
-
image_embedding = gr.Interface(fn=get_image_embedding, inputs=gr.Image(type="
|
20 |
text_embedding = gr.Interface(fn=get_text_embedding, inputs=gr.Textbox(), outputs=gr.JSON(api_name="text-embedding"), title="Text Embedding")
|
21 |
|
22 |
space = gr.TabbedInterface([image_embedding, text_embedding], ["Image Embedding", "Text Embedding"])
|
|
|
16 |
print(text_emb.ndim)
|
17 |
return {"embedding": text_emb.tolist()}
|
18 |
|
19 |
+
image_embedding = gr.Interface(fn=get_image_embedding, inputs=gr.Image(type="pil"), outputs=gr.JSON(api_name="image-embedding"), title="Image Embedding")
|
20 |
text_embedding = gr.Interface(fn=get_text_embedding, inputs=gr.Textbox(), outputs=gr.JSON(api_name="text-embedding"), title="Text Embedding")
|
21 |
|
22 |
space = gr.TabbedInterface([image_embedding, text_embedding], ["Image Embedding", "Text Embedding"])
|