cblock commited on
Commit
cc1239c
·
verified ·
1 Parent(s): 97b22b4

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.

Files changed (1) hide show
  1. app.py +6 -6
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', use_auth_token=str(os.environ['hf_token']))['train'])
13
- corpus_eng = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_eng', use_auth_token=str(os.environ['hf_token']))['train'])
14
- corpus_fr = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_fr', use_auth_token=str(os.environ['hf_token']))['train'])
15
 
16
  #Import models
17
- model_ger = SentenceTransformer('ECLASS-Standard/gbert-base-eclass', use_auth_token=str(os.environ['hf_token']))
18
- model_eng = SentenceTransformer('ECLASS-Standard/mboth-distil-eng-quora-sentence', use_auth_token=str(os.environ['hf_token']))
19
- model_fr = SentenceTransformer('ECLASS-Standard/Sahajtomar-french_semantic', use_auth_token=str(os.environ['hf_token']))
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):