Skip to content

Commit b5ef811

Browse files
committed
Explicit path and model for bertopic
1 parent c820099 commit b5ef811

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ dependencies = [
2121
"tiktoken>=0.5.1",
2222
"numpy>=2.1.0",
2323
"pydantic-settings>=2.8.1",
24-
"bertopic>=0.16.4",
24+
"bertopic>>0.16.4 <=0.17.0",
2525
"structlog>=25.2.0",
2626
"transformers>=4.30.0",
2727
"numba>=0.60.0",

redis_memory_server/extraction.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def get_topic_model() -> BERTopic:
2828
"""
2929
global _topic_model
3030
if _topic_model is None:
31-
_topic_model = BERTopic.load(settings.topic_model)
31+
_topic_model = BERTopic.load("topic_model.pickle", settings.topic_model)
3232
return _topic_model # type: ignore
3333

3434

0 commit comments

Comments
 (0)