Update app.py

app.py

@@ -42,13 +42,13 @@ def get_embeddings(input):
 
 # Load data
 ds_name = 'AresEkb/prof_standards_sbert_large_mt_nlu_ru'
-domains_ds = load_dataset(ds_name, 'domains').add_faiss_index(column='embeddings')
-generalized_functions_ds = load_dataset(ds_name, 'generalized_functions').add_faiss_index(column='embeddings')
-jobs_ds = load_dataset(ds_name, 'jobs').add_faiss_index(column='embeddings')
-particular_functions_ds = load_dataset(ds_name, 'particular_functions').add_faiss_index(column='embeddings')
-actions_ds = load_dataset(ds_name, 'actions').add_faiss_index(column='embeddings')
-skills_ds = load_dataset(ds_name, 'skills').add_faiss_index(column='embeddings')
-knowledges_ds = load_dataset(ds_name, 'knowledges').add_faiss_index(column='embeddings')
+domains_ds = load_dataset(ds_name, 'domains', split='train').add_faiss_index(column='embeddings')
+generalized_functions_ds = load_dataset(ds_name, 'generalized_functions', split='train').add_faiss_index(column='embeddings')
+jobs_ds = load_dataset(ds_name, 'jobs', split='train').add_faiss_index(column='embeddings')
+particular_functions_ds = load_dataset(ds_name, 'particular_functions', split='train').add_faiss_index(column='embeddings')
+actions_ds = load_dataset(ds_name, 'actions', split='train').add_faiss_index(column='embeddings')
+skills_ds = load_dataset(ds_name, 'skills', split='train').add_faiss_index(column='embeddings')
+knowledges_ds = load_dataset(ds_name, 'knowledges', split='train').add_faiss_index(column='embeddings')
 
 indices = {'reg_number', 'generalized_function_id', 'particular_function_id'}