Hugging Face's logo

Spaces:
harshithakr
/
mapping_bert_topic
Runtime error

App Files Community
harshithakr commited on
Commit
bf78592
·
1 Parent(s): 241a7ed

Update mapping.py

Browse files
Files changed (1) hide show
  1. mapping.py +1 -1
mapping.py CHANGED
@@ -6,7 +6,7 @@ model_sent = SentenceTransformer("all-mpnet-base-v2")
6
 
7
  sector_model = pickle.load(open('sector_knn.sav', 'rb'))
8
  indus_model = pickle.load(open('indus_knn.sav', 'rb'))
9
- other_indus_model = pickle.load(open('other_industries.sav', 'rb'))
10
 
11
  def get_mapping(prep_text):
12
  event_discr_embeddings = model_sent.encode([preprocess_text(prep_text)],
 
6
 
7
  sector_model = pickle.load(open('sector_knn.sav', 'rb'))
8
  indus_model = pickle.load(open('indus_knn.sav', 'rb'))
9
+ other_indus_model = pickle.load(open('other_industries_knn.sav', 'rb'))
10
 
11
  def get_mapping(prep_text):
12
  event_discr_embeddings = model_sent.encode([preprocess_text(prep_text)],