Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -9,9 +9,9 @@ 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 |
#Import models
|
16 |
model_ger = SentenceTransformer('ECLASS-Standard/gbert-base-eclass') #, token=str(os.environ['private_token'])
|
17 |
model_eng = SentenceTransformer('ECLASS-Standard/mboth-distil-eng-quora-sentence') #, token=str(os.environ['private_token'])
|
|
|
9 |
import os
|
10 |
|
11 |
#Import corpus embeddings
|
12 |
+
corpus_ger = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_ger', token=str(os.environ['private_token'])))['train']
|
13 |
+
corpus_eng = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_eng', token=str(os.environ['private_token'])))['train']
|
14 |
+
corpus_fr = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_fr', token=str(os.environ['private_token'])))['train']
|
15 |
#Import models
|
16 |
model_ger = SentenceTransformer('ECLASS-Standard/gbert-base-eclass') #, token=str(os.environ['private_token'])
|
17 |
model_eng = SentenceTransformer('ECLASS-Standard/mboth-distil-eng-quora-sentence') #, token=str(os.environ['private_token'])
|