Update model_utils.py
Browse files- model_utils.py +2 -0
model_utils.py
CHANGED
|
@@ -35,6 +35,8 @@ embed_model = SentenceTransformer(EMBED_MODEL_NAME)
|
|
| 35 |
embed_model = embed_model.to(device)
|
| 36 |
|
| 37 |
|
|
|
|
|
|
|
| 38 |
def _build_entry_embeddings() -> None:
|
| 39 |
"""
|
| 40 |
Build embeddings for each textbook entry using title + summary + text
|
|
|
|
| 35 |
embed_model = embed_model.to(device)
|
| 36 |
|
| 37 |
|
| 38 |
+
# NOTE: called once after load_curriculum() to precompute embeddings.
|
| 39 |
+
# If you ever reload curriculum at runtime, call _build_entry_embeddings() again.
|
| 40 |
def _build_entry_embeddings() -> None:
|
| 41 |
"""
|
| 42 |
Build embeddings for each textbook entry using title + summary + text
|