Update app.py

app.py

@@ -37,15 +37,19 @@ def predict(text):
     sim_scores1 = util.pytorch_cos_sim(query_embedding, embeddings)[0]
     sim_scores2 = util.pytorch_cos_sim(query_embedding, embeddings2)[0]
     sim_scores3 = util.pytorch_cos_sim(query_embedding, embeddings3)[0]
+    
     all_sim_scores1.append(sim_scores1)
     all_sim_scores2.append(sim_scores2)
     all_sim_scores1.append(sim_scores3)
-    sim_scores1 = util.pytorch_cos_sim(query_embeddinga, embeddingsa)[0]
-    sim_scores2 = util.pytorch_cos_sim(query_embeddinga, embeddingsa2)[0]
-    sim_scores3 = util.pytorch_cos_sim(query_embeddinga, embeddingsa3)[0]
-    all_sim_scores1.append(sim_scores1)
-    all_sim_scores2.append(sim_scores2)
-    all_sim_scores1.append(sim_scores3)
+    
+    sim_scores1a = util.pytorch_cos_sim(query_embeddinga, embeddingsa)[0]
+    sim_scores2a = util.pytorch_cos_sim(query_embeddinga, embeddingsa2)[0]
+    sim_scores3a = util.pytorch_cos_sim(query_embeddinga, embeddingsa3)[0]
+    
+    all_sim_scores1.append(sim_scores1a)
+    all_sim_scores2.append(sim_scores2a)
+    all_sim_scores1.append(sim_scores3a)
+    
     sim_scores1 = torch.stack(all_sim_scores1).mean(dim=0)
     sim_scores2 = torch.stack(all_sim_scores2).mean(dim=0)
     sim_scores3 = torch.stack(all_sim_scores3).mean(dim=0)