gschurck commited on
Commit
2bcb84c
·
1 Parent(s): 7eb7790

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="url"), 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"])
 
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"])