Spaces:
Runtime error
Runtime error
Commit
·
6b1623b
1
Parent(s):
f4d66aa
Update segmentation.py
Browse files- segmentation.py +0 -2
segmentation.py
CHANGED
|
@@ -7,7 +7,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'):
|
| 12 |
"""
|
| 13 |
all_MiniLM_L6_v2 - offline
|
|
@@ -16,7 +15,6 @@ def load_sentence_transformer(model_name='all-MiniLM-L6-v2'):
|
|
| 16 |
model = SentenceTransformer(model_name)
|
| 17 |
return model
|
| 18 |
|
| 19 |
-
@lru_cache
|
| 20 |
def load_spacy():
|
| 21 |
return spacy.load('en_core_web_sm')
|
| 22 |
|
|
|
|
| 7 |
from functools import lru_cache
|
| 8 |
from sentence_transformers import SentenceTransformer
|
| 9 |
|
|
|
|
| 10 |
def load_sentence_transformer(model_name='all-MiniLM-L6-v2'):
|
| 11 |
"""
|
| 12 |
all_MiniLM_L6_v2 - offline
|
|
|
|
| 15 |
model = SentenceTransformer(model_name)
|
| 16 |
return model
|
| 17 |
|
|
|
|
| 18 |
def load_spacy():
|
| 19 |
return spacy.load('en_core_web_sm')
|
| 20 |
|