gschurck commited on
Commit
1ec0c8c
·
1 Parent(s): 6ff5474

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,8 +13,8 @@ def get_text_embedding(text):
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="pil"), outputs=gr.JSON(), title="Image Embedding")
17
- text_embedding = gr.Interface(fn=get_text_embedding, inputs=gr.Textbox(), outputs=gr.JSON(), title="Text Embedding")
18
 
19
  space = gr.TabbedInterface([image_embedding, text_embedding], ["Image Embedding", "Text Embedding"])
20
  space.launch()
 
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="pil"), outputs=gr.JSON(api_name="image-embedding"), title="Image Embedding")
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"])
20
  space.launch()