Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -9,10 +9,10 @@ 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', token=str(os.environ['private_token']))
|
18 |
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']))
|
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']))
|