AkashKhamkar commited on
Commit
edef549
·
1 Parent(s): bad3815

Update segmentation.py

Browse files
Files changed (1) hide show
  1. 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 SentenceTransformers
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'):