Spaces:
Running
Running
disable token
Browse files
app.py
CHANGED
@@ -9,14 +9,14 @@ from datasets import load_dataset
|
|
9 |
import os
|
10 |
|
11 |
#Import corpus embeddings
|
12 |
-
corpus_ger = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_ger'
|
13 |
-
corpus_eng = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_eng'
|
14 |
-
corpus_fr = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_fr'
|
15 |
|
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):
|
|
|
9 |
import os
|
10 |
|
11 |
#Import corpus embeddings
|
12 |
+
corpus_ger = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_ger') #, token=str(os.environ['hf_token']))['train']
|
13 |
+
corpus_eng = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_eng') #, token=str(os.environ['hf_token']))['train']
|
14 |
+
corpus_fr = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_fr') #, token=str(os.environ['hf_token']))['train']
|
15 |
|
16 |
#Import models
|
17 |
+
model_ger = SentenceTransformer('ECLASS-Standard/gbert-base-eclass') #, token=str(os.environ['hf_token'])
|
18 |
+
model_eng = SentenceTransformer('ECLASS-Standard/mboth-distil-eng-quora-sentence') #, token=str(os.environ['hf_token'])
|
19 |
+
model_fr = SentenceTransformer('ECLASS-Standard/Sahajtomar-french_semantic') #, token=str(os.environ['hf_token'])
|
20 |
|
21 |
#Definition of search function
|
22 |
def predict(name, description, language, classCode='nofilter', top_k=10):
|