Hugging Face's logo

Spaces:
ECLASS-Standard
/
ECLASS-Search
Sleeping

App Files Community
2
JoBeer commited on
Commit
7c1a47f
·
verified ·
1 Parent(s): feff232

Update app.py

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