Update app.py

app.py

@@ -11,10 +11,15 @@ def get_image_embedding(image):
     img_emb = image_model.encode(image)
     print(img_emb)
     print(type(img_emb))
-    return {"prediction": img_emb.tolist()}
+    return {"embedding": img_emb.tolist()}
 
+def get_text_embedding(text):
+    multilingual_text_model = SentenceTransformer('clip-ViT-B-32-multilingual-v1')
+    text_emb = multilingual_text_model.encode(text)
+    return {"embedding": text_emb}
 
-image_input = gr.Image(type="pil")
-label_output = gr.JSON()
+image_embedding = gr.Interface(fn=get_image_embedding, inputs=gr.Image(type="pil"), outputs=gr.JSON())
+text_embedding = gr.Interface(fn=get_text_embedding, inputs=gr.Textbox(), outputs=gr.JSON())
 
-gr.Interface(fn=get_image_embedding, inputs=image_input, outputs=label_output).launch()
+space = gr.TabbedInterface([image_embedding, text_embedding], ["Image Embedding", "Text Embedding"])
+space.launch()