liuwei commited on
Commit
15936c4
·
1 Parent(s): a91b3c4
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def app():
40
  sent_embs = model.encode(sentences, convert_to_tensor=True)
41
  cos_sim = util.cos_sim(source_emb, sent_embs)
42
 
43
- cosin_dict = []
44
 
45
  for i, cos in cos_sim:
46
  cosin_dict[sentences[i]] = cos
 
40
  sent_embs = model.encode(sentences, convert_to_tensor=True)
41
  cos_sim = util.cos_sim(source_emb, sent_embs)
42
 
43
+ cosin_dict = {}
44
 
45
  for i, cos in cos_sim:
46
  cosin_dict[sentences[i]] = cos