JoBeer commited on
Commit
96664a2
·
1 Parent(s): 116839d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ corpus = pd.read_excel('corpus.xlsx')
13
  def predict(name, description):
14
  text = 'Description: '+ description + '; Name: ' + name
15
  query_embedding = model.encode(text, convert_to_tensor=True)
16
-
17
  corpus_embeddings = torch.Tensor(corpus["embeddings"])
18
 
19
  output = sentence_transformers.util.semantic_search(query_embedding, corpus_embeddings, top_k = 5)
 
13
  def predict(name, description):
14
  text = 'Description: '+ description + '; Name: ' + name
15
  query_embedding = model.encode(text, convert_to_tensor=True)
16
+
17
  corpus_embeddings = torch.Tensor(corpus["embeddings"])
18
 
19
  output = sentence_transformers.util.semantic_search(query_embedding, corpus_embeddings, top_k = 5)