Spaces:
Runtime error
Runtime error
Commit ·
2cad3f7
1
Parent(s): 4e6f134
Update mapping.py
Browse files- 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]
|