Yeetek commited on
Commit
3133525
Β·
verified Β·
1 Parent(s): 54f370c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = SentenceTransformer(MODEL_NAME)
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,