Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -56,7 +56,7 @@ MIN_TOPIC = int(os.getenv("MIN_TOPIC_SIZE", "10"))
|
|
56 |
MAX_DOCS = int(os.getenv("MAX_DOCS", "5000"))
|
57 |
|
58 |
# ββ 5. Initialise models once at container start βββββββββββββββββββββββββββββ
|
59 |
-
embeddings
|
60 |
topic_model = BERTopic(
|
61 |
embedding_model=embeddings,
|
62 |
min_topic_size=MIN_TOPIC,
|
|
|
56 |
MAX_DOCS = int(os.getenv("MAX_DOCS", "5000"))
|
57 |
|
58 |
# ββ 5. Initialise models once at container start βββββββββββββββββββββββββββββ
|
59 |
+
embeddings = SentenceTransformer(MODEL_NAME, cache_folder="/tmp/hfcache")
|
60 |
topic_model = BERTopic(
|
61 |
embedding_model=embeddings,
|
62 |
min_topic_size=MIN_TOPIC,
|