Yeetek commited on
Commit
98c2919
Β·
verified Β·
1 Parent(s): 14beccb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -1,10 +1,15 @@
1
- # ── DIAGNOSTICS (top of file) ─────────────────────────────────────────────
2
- import pkgutil, sentence_transformers, bertopic, sys, json
3
  print("ST version:", sentence_transformers.__version__)
4
  print("BERTopic version:", bertopic.__version__)
5
- print("ST model modules:", [m.name for m in pkgutil.iter_modules(sentence_transformers.models.__path__)])
6
  sys.stdout.flush()
7
- # ─────────────────────────────────────────────────────────────────────────
 
 
 
 
 
8
 
9
  # ── regular imports and monkey-patch if you choose ───────────────────────
10
  from bertopic import BERTopic
 
1
+ # ── DIAGNOSTICS ─────────────────────────────────────────────
2
+ import pkgutil, sentence_transformers, bertopic, sys
3
  print("ST version:", sentence_transformers.__version__)
4
  print("BERTopic version:", bertopic.__version__)
5
+ print("ST models:", [m.name for m in pkgutil.iter_modules(sentence_transformers.models.__path__)])
6
  sys.stdout.flush()
7
+
8
+ # ── STATICEMBEDDING SHIM ────────────────────────────────────
9
+ if "StaticEmbedding" not in [m.name for m in pkgutil.iter_modules(sentence_transformers.models.__path__)]:
10
+ from sentence_transformers.models import Transformer
11
+ import bertopic.backend._sentencetransformers as st_back
12
+ st_back.StaticEmbedding = Transformer
13
 
14
  # ── regular imports and monkey-patch if you choose ───────────────────────
15
  from bertopic import BERTopic