sergiomar73 commited on
Commit
0d4f610
1 Parent(s): d85f21f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ def compare_text(source, target):
13
  embeddings_target = model.encode(target, convert_to_tensor=True)
14
  # Compute cosine-similarities
15
  cosine_scores = util.cos_sim(embeddings_source, embeddings_target)
16
- return { "Similarity": cosine_scores[0][0] }
17
 
18
  import gradio as gr
19
 
 
13
  embeddings_target = model.encode(target, convert_to_tensor=True)
14
  # Compute cosine-similarities
15
  cosine_scores = util.cos_sim(embeddings_source, embeddings_target)
16
+ return { "Similarity": cosine_scores[0][0].item() }
17
 
18
  import gradio as gr
19