JoBeer commited on
Commit
4b4d18a
·
verified ·
1 Parent(s): 56b2847

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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'))['train'] #, token=str(os.environ['private_token'])
13
- corpus_eng = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_eng'))['train'] #, token=str(os.environ['private_token'])
14
- corpus_fr = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_fr'))['train'] #, token=str(os.environ['private_token'])
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'])