Spaces:
Running
Running
Update app.py
Browse files/home/user/.local/lib/python3.10/site-packages/datasets/load.py:2483: FutureWarning: 'use_auth_token' was deprecated in favor of 'token' in version 2.14.0 and will be removed in 3.0.0.
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):
|