jonathanjordan21 commited on
Commit
64f2cc4
·
verified ·
1 Parent(s): 5bd1dac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -511,7 +511,7 @@ def reload(chosen_model_id):
511
  model = SentenceTransformer(chosen_model_id, trust_remote_code=True)
512
  model_id = chosen_model_id
513
  # codes_emb = model.encode([x[6:] for x in codes])
514
- codes_emb = model.encode([x["examples"] for x in examples]).mean(axis=1)
515
  return f"Model {chosen_model_id} has been succesfully loaded!"
516
  return f"Model {chosen_model_id} is ready!"
517
 
 
511
  model = SentenceTransformer(chosen_model_id, trust_remote_code=True)
512
  model_id = chosen_model_id
513
  # codes_emb = model.encode([x[6:] for x in codes])
514
+ codes_emb = model.encode([x["examples"] for x in examples])#.mean(axis=1)
515
  return f"Model {chosen_model_id} has been succesfully loaded!"
516
  return f"Model {chosen_model_id} is ready!"
517