Update app.py

app.py

@@ -13,12 +13,13 @@ def compare_text(source, target):
   embeddings_target = model.encode(target, convert_to_tensor=True)
   # Compute cosine-similarities
   cosine_scores = util.cos_sim(embeddings_source, embeddings_target)
-  return { "Similarity": cosine_scores[0][0].item() }
+  cosine = cosine_scores[0][0].item() 
+  return { "Similarity": cosine }, round(cosine,3)
 
 import gradio as gr
 
 with gr.Blocks(css=".gradio-container { background-color: white; background-image: url('file=./qc-logo.png'); background-size: 75px 75px; background-repeat: no-repeat; background-position: 0px 0px; }") as demo:
-    gr.Markdown(f"# {' ' * 8}Sentence comparer with Roberta")
+    gr.Markdown(f"# {' ' * 8}Sentence comparer with BERT (paraphrase-distilroberta-base-v2)")
     with gr.Row():
         source = gr.Textbox(lines=3, label="Source sentence", placeholder="Source sentence")
     with gr.Row():
@@ -26,7 +27,8 @@ with gr.Blocks(css=".gradio-container { background-color: white; background-imag
     btn = gr.Button(value="Compare!", variant="primary")
     with gr.Row():
         label = gr.Label()
-    btn.click(fn=compare_text, inputs=[source,target], outputs=[label])
+        score = gr.Textbox(label="Score", interactive=False)
+    btn.click(fn=compare_text, inputs=[source,target], outputs=[label,score])
     gr.Examples([
         ["Oh, so the quantified platform is one of the most advanced communication intelligence in AI powered coaching systems.","most advanced conversation intelligence and AI powered coaching platform"],
         ["Oh, so the quantified platform is one of the most advanced communication intelligence in AI powered coaching systems. And what does that really mean? So, um, communication coaching is something that is typically delivered one on one between a communication coach who has a, uh, a doctorate or a, um, background and experience in teaching people how to be better communicators and how to express themselves effectively.","most advanced conversation intelligence and AI powered coaching platform"],