JoBeer commited on
Commit
3c880d0
·
verified ·
1 Parent(s): b975d7b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,9 +14,9 @@ corpus_eng = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_eng',
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']))
19
- model_fr = SentenceTransformer('ECLASS-Standard/Sahajtomar-french_semantic', token=str(os.environ['private_token']))
20
 
21
  #Definition of search function
22
  def predict(name, description, language, classCode='nofilter', top_k=10):
 
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'])
19
+ model_fr = SentenceTransformer('ECLASS-Standard/Sahajtomar-french_semantic') #, token=str(os.environ['private_token'])
20
 
21
  #Definition of search function
22
  def predict(name, description, language, classCode='nofilter', top_k=10):