Spaces:
Running
Running
Update app.py
Browse files
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
|
19 |
-
model_fr = SentenceTransformer('Sahajtomar
|
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):
|