Update app.py

app.py

@@ -605,8 +605,8 @@ def reload(chosen_model_id):
         model = SentenceTransformer(chosen_model_id, trust_remote_code=True)
         model_id = chosen_model_id
         # 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)
         return f"Model {chosen_model_id} has been succesfully loaded!"
     return f"Model {chosen_model_id} is ready!"