jeffrey1963 commited on
Commit
ba04b68
·
verified ·
1 Parent(s): 54e7208

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -1,20 +1,20 @@
1
  import gradio as gr
2
  from sentence_transformers import SentenceTransformer, util
3
 
4
- # Load sentence embedding model
5
- model = SentenceTransformer("all-MiniLM-L6-v2")
6
 
7
  def compare_sentences(sentence_a, sentence_b):
8
  emb = model.encode([sentence_a, sentence_b], convert_to_tensor=True)
9
  score = util.pytorch_cos_sim(emb[0], emb[1]).item()
10
- return f"{score * 100:.2f}%" # Convert to percentage
11
 
12
  iface = gr.Interface(
13
  fn=compare_sentences,
14
  inputs=["text", "text"],
15
  outputs="text",
16
- title="Sentence Similarity Checker",
17
- description="Enter two sentences to compare their similarity (0–100%) using BERT embeddings."
18
  )
19
 
20
  iface.launch()
 
1
  import gradio as gr
2
  from sentence_transformers import SentenceTransformer, util
3
 
4
+ # Use a more powerful model for paraphrasing and semantic similarity
5
+ model = SentenceTransformer("sentence-transformers/all-mpnet-base-v2")
6
 
7
  def compare_sentences(sentence_a, sentence_b):
8
  emb = model.encode([sentence_a, sentence_b], convert_to_tensor=True)
9
  score = util.pytorch_cos_sim(emb[0], emb[1]).item()
10
+ return f"{score * 100:.2f}%" # Similarity score as percentage
11
 
12
  iface = gr.Interface(
13
  fn=compare_sentences,
14
  inputs=["text", "text"],
15
  outputs="text",
16
+ title="Sentence Similarity Checker (MPNet Model)",
17
+ description="Enter two sentences to compare their meaning using a powerful transformer model. Score is shown as a percent."
18
  )
19
 
20
  iface.launch()