Spaces:
Runtime error
Runtime error
Commit
·
edef549
1
Parent(s):
bad3815
Update segmentation.py
Browse files- segmentation.py +1 -1
segmentation.py
CHANGED
|
@@ -5,7 +5,7 @@ import spacy
|
|
| 5 |
from nltk.tokenize.texttiling import TextTilingTokenizer
|
| 6 |
from sklearn.metrics.pairwise import cosine_similarity
|
| 7 |
from functools import lru_cache
|
| 8 |
-
from sentence_transformers import
|
| 9 |
|
| 10 |
@lru_cache
|
| 11 |
def load_sentence_transformer(model_name='all_MiniLM_L6_v2'):
|
|
|
|
| 5 |
from nltk.tokenize.texttiling import TextTilingTokenizer
|
| 6 |
from sklearn.metrics.pairwise import cosine_similarity
|
| 7 |
from functools import lru_cache
|
| 8 |
+
from sentence_transformers import SentenceTransformer
|
| 9 |
|
| 10 |
@lru_cache
|
| 11 |
def load_sentence_transformer(model_name='all_MiniLM_L6_v2'):
|