Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -14,9 +14,9 @@ corpus_eng = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_eng',
|
|
14 |
corpus_fr = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_fr', token=str(os.environ['private_token']))
|
15 |
)['train']
|
16 |
#Import models
|
17 |
-
model_ger = SentenceTransformer('ECLASS-Standard/gbert-base-eclass'
|
18 |
-
model_eng = SentenceTransformer('ECLASS-Standard/mboth-distil-eng-quora-sentence'
|
19 |
-
model_fr = SentenceTransformer('ECLASS-Standard/Sahajtomar-french_semantic'
|
20 |
|
21 |
#Definition of search function
|
22 |
def predict(name, description, language, classCode='nofilter', top_k=10):
|
|
|
14 |
corpus_fr = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_fr', token=str(os.environ['private_token']))
|
15 |
)['train']
|
16 |
#Import models
|
17 |
+
model_ger = SentenceTransformer('ECLASS-Standard/gbert-base-eclass') #, token=str(os.environ['private_token'])
|
18 |
+
model_eng = SentenceTransformer('ECLASS-Standard/mboth-distil-eng-quora-sentence') #, token=str(os.environ['private_token'])
|
19 |
+
model_fr = SentenceTransformer('ECLASS-Standard/Sahajtomar-french_semantic') #, token=str(os.environ['private_token'])
|
20 |
|
21 |
#Definition of search function
|
22 |
def predict(name, description, language, classCode='nofilter', top_k=10):
|