Update app.py

app.py

@@ -305,8 +305,8 @@ model_id = model_ids[-1]
 model = SentenceTransformer(model_id, trust_remote_code=True)
 
 # codes_emb = model.encode([x[6:] for x in codes])
-# codes_emb = model.encode([x["examples"] for x in examples])#.mean(axis=1)
-codes_emb = np.mean([model.encode(x["examples"]) for x in examples], axis=1)
+codes_emb = model.encode([x["examples"] for x in examples])#.mean(axis=1)
+# codes_emb = np.mean([model.encode(x["examples"]) for x in examples], axis=1)
 # for x in examples:
 #   codes_emb.append(model.encode(x["examples"]))
 # codes_emb = np.mean(codes_emb, axis=1)