nitinbhayana commited on
Commit
ef5a693
1 Parent(s): 063e402

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -20,7 +20,11 @@ def predict(text):
20
  term_embedding = term_outputs.mean(dim=1)
21
  #term_embedding = term_outputs.last_hidden_state.mean(dim=1)
22
  semantic_score = cosine_similarity(title_embedding.flatten(), term_embedding.flatten()) * 100
23
- return str(format(semantic_score,'.2f'))
 
 
 
 
24
  def predict1(text):
25
  return(text)
26
 
 
20
  term_embedding = term_outputs.mean(dim=1)
21
  #term_embedding = term_outputs.last_hidden_state.mean(dim=1)
22
  semantic_score = cosine_similarity(title_embedding.flatten(), term_embedding.flatten()) * 100
23
+ return title_embedding #str(format(semantic_score,'.2f'))
24
+
25
+
26
+
27
+
28
  def predict1(text):
29
  return(text)
30