Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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])
|
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 |
|