Update app.py

app.py

@@ -1,15 +1,20 @@
 import gradio as gr
-
-#gr.load("models/WhereIsAI/UAE-Large-V1").launch()
-
-# Use a pipeline as a high-level helper
 from transformers import pipeline
 
-pipe = pipeline("feature-extraction", model="WhereIsAI/UAE-Large-V1")
+pipeline = pipeline("feature-extraction", model="WhereIsAI/UAE-Large-V1")
 
-gr.Interface(
+def similarity(text):
+    title_embedding = pipeline(text)
+    term_embedding = pipeline('multivitamin for men')
+    semantic_score = cosine_similarity(title_embedding.flatten(), term_embedding.flatten()) * 100
+    return str(format(semantic_score,'.2f'))
+
+gradio_app = gr.Interface(
     predict,
     inputs='text',
     outputs='text',
-    title="Mapping",
-).launch()
+    title="Yes Or No?",
+)
+
+if __name__ == "__main__":
+    gradio_app.launch()