cools commited on
Commit
78387c0
·
1 Parent(s): a281f5b

Update Tagger.py

Browse files
Files changed (1) hide show
  1. Tagger.py +2 -0
Tagger.py CHANGED
@@ -5,6 +5,8 @@ import re
5
  from sentence_transformers import SentenceTransformer
6
  from sentence_transformers.util import cos_sim
7
  import cv2
 
 
8
  from nltk.tokenize import sent_tokenize, word_tokenize
9
 
10
  model = SentenceTransformer('all-mpnet-base-v2')
 
5
  from sentence_transformers import SentenceTransformer
6
  from sentence_transformers.util import cos_sim
7
  import cv2
8
+ import nltk
9
+ nltk.download('punkt')
10
  from nltk.tokenize import sent_tokenize, word_tokenize
11
 
12
  model = SentenceTransformer('all-mpnet-base-v2')