JoBeer commited on
Commit
97b22b4
·
verified ·
1 Parent(s): 7c9143a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,8 +15,8 @@ corpus_fr = pd.DataFrame(load_dataset('ECLASS-Standard/eclass_properties_fr', us
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('mboth/distil-eng-quora-sentence')
19
- model_fr = SentenceTransformer('Sahajtomar/french_semantic')
20
 
21
  #Definition of search function
22
  def predict(name, description, language, classCode='nofilter', top_k=10):
 
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):