mohbay commited on
Commit
7d96acd
·
verified ·
1 Parent(s): c0ac38e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,7 +40,7 @@ def predict(text):
40
 
41
  all_sim_scores1.append(sim_scores1)
42
  all_sim_scores2.append(sim_scores2)
43
- all_sim_scores1.append(sim_scores3)
44
 
45
  sim_scores1a = util.pytorch_cos_sim(query_embeddinga, embeddingsa)[0]
46
  sim_scores2a = util.pytorch_cos_sim(query_embeddinga, embeddingsa2)[0]
@@ -48,7 +48,7 @@ def predict(text):
48
 
49
  all_sim_scores1.append(sim_scores1a)
50
  all_sim_scores2.append(sim_scores2a)
51
- all_sim_scores1.append(sim_scores3a)
52
 
53
  sim_scores1 = torch.stack(all_sim_scores1).mean(dim=0)
54
  sim_scores2 = torch.stack(all_sim_scores2).mean(dim=0)
 
40
 
41
  all_sim_scores1.append(sim_scores1)
42
  all_sim_scores2.append(sim_scores2)
43
+ all_sim_scores3.append(sim_scores3)
44
 
45
  sim_scores1a = util.pytorch_cos_sim(query_embeddinga, embeddingsa)[0]
46
  sim_scores2a = util.pytorch_cos_sim(query_embeddinga, embeddingsa2)[0]
 
48
 
49
  all_sim_scores1.append(sim_scores1a)
50
  all_sim_scores2.append(sim_scores2a)
51
+ all_sim_scores3.append(sim_scores3a)
52
 
53
  sim_scores1 = torch.stack(all_sim_scores1).mean(dim=0)
54
  sim_scores2 = torch.stack(all_sim_scores2).mean(dim=0)