Update app.py

app.py

@@ -1,23 +1,29 @@
-from sentence_transformers import SentenceTransformer, util
-from PIL import Image
+from sentence_transformers import SentenceTransformer
 import gradio as gr
-import requests
+
+# Load once
+image_model = SentenceTransformer("clip-ViT-B-32")
+text_model = SentenceTransformer("clip-ViT-B-32-multilingual-v1")
 
 def get_image_embedding(image):
-    image_model = SentenceTransformer('clip-ViT-B-32')
-    img_emb = image_model.encode(image)
-    return {"embedding": img_emb.tolist()}
+    emb = image_model.encode(image)
+    return {"embedding": emb.tolist()}
 
 def get_text_embedding(text):
-    multilingual_text_model = SentenceTransformer('clip-ViT-B-32-multilingual-v1')
-    text_emb = multilingual_text_model.encode(text)
-    print(text_emb)
-    print(type(text_emb))
-    print(text_emb.ndim)
-    return {"embedding": text_emb.tolist()}
+    emb = text_model.encode(text)
+    return {"embedding": emb.tolist()}
+
+with gr.Blocks() as demo:
+    with gr.Tab("Image Embedding"):
+        img_input = gr.Image(type="pil")
+        img_output = gr.JSON()
+        img_btn = gr.Button("Generate")
+        img_btn.click(get_image_embedding, img_input, img_output)
 
-image_embedding = gr.Interface(fn=get_image_embedding, inputs=gr.Image(type="pil"), outputs=gr.JSON(api_name="image-embedding"), title="Image Embedding")
-text_embedding = gr.Interface(fn=get_text_embedding, inputs=gr.Textbox(), outputs=gr.JSON(api_name="text-embedding"), title="Text Embedding")
+    with gr.Tab("Text Embedding"):
+        text_input = gr.Textbox()
+        text_output = gr.JSON()
+        text_btn = gr.Button("Generate")
+        text_btn.click(get_text_embedding, text_input, text_output)
 
-space = gr.TabbedInterface([image_embedding, text_embedding], ["Image Embedding", "Text Embedding"])
-space.launch()
+demo.launch()