harshithakr commited on
Commit
2cad3f7
·
1 Parent(s): 4e6f134

Update mapping.py

Browse files
Files changed (1) hide show
  1. mapping.py +2 -1
mapping.py CHANGED
@@ -1,12 +1,13 @@
1
  import pickle
2
  from sentence_transformers import SentenceTransformer, util
 
3
  model_sent = SentenceTransformer("all-mpnet-base-v2")
4
 
5
  sector_model = pickle.load(open('/main/sector_knn.sav', 'rb'))
6
  other_indus_model = pickle.load(open('/main/other_indus_knn.sav', 'rb'))
7
 
8
  def get_mapping(prep_text):
9
- event_discr_embeddings = model_sent.encode([prep_text],
10
  batch_size=250,
11
  show_progress_bar=True)
12
  event_embedd = event_discr_embeddings[0]
 
1
  import pickle
2
  from sentence_transformers import SentenceTransformer, util
3
+ from preprocess_function import preprocess_text
4
  model_sent = SentenceTransformer("all-mpnet-base-v2")
5
 
6
  sector_model = pickle.load(open('/main/sector_knn.sav', 'rb'))
7
  other_indus_model = pickle.load(open('/main/other_indus_knn.sav', 'rb'))
8
 
9
  def get_mapping(prep_text):
10
+ event_discr_embeddings = model_sent.encode([preprocess_text(prep_text)],
11
  batch_size=250,
12
  show_progress_bar=True)
13
  event_embedd = event_discr_embeddings[0]