etadevosyan commited on
Commit
3bdf307
·
1 Parent(s): 18ed8b2
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -14,11 +14,11 @@ data = prepare_data()
14
  HF_TOKEN = os.getenv('HF_TOKEN')
15
  hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "budu_search_data_new")
16
 
17
- new = Dataset.from_dict({'Введите запрос':[],
18
- 'output':[]})
19
  new.push_to_hub("etadevosyan/not_flagged_data",
20
  token=HF_TOKEN)
21
-
22
  def search_handler(query: str):
23
  results, exact_results = (
24
  search_bm25(query, data),
@@ -38,7 +38,7 @@ def create_ui(query):
38
 
39
  def log_non_flagged_queries(query,recommendations):
40
  dataset = load_dataset('etadevosyan/not_flagged_data')
41
- dataset.add_item({'Введите запрос':query,
42
  'output':recommendations})
43
  dataset.push_to_hub("etadevosyan/not_flagged_data",
44
  token=HF_TOKEN)
 
14
  HF_TOKEN = os.getenv('HF_TOKEN')
15
  hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "budu_search_data_new")
16
 
17
+ new = Dataset.from_dict({'Введите запрос':['тест'],
18
+ 'output':['тест']},split='train')
19
  new.push_to_hub("etadevosyan/not_flagged_data",
20
  token=HF_TOKEN)
21
+ #%%
22
  def search_handler(query: str):
23
  results, exact_results = (
24
  search_bm25(query, data),
 
38
 
39
  def log_non_flagged_queries(query,recommendations):
40
  dataset = load_dataset('etadevosyan/not_flagged_data')
41
+ dataset['train'].add_item({'Введите запрос':query,
42
  'output':recommendations})
43
  dataset.push_to_hub("etadevosyan/not_flagged_data",
44
  token=HF_TOKEN)