Update app.py

app.py

@@ -5,6 +5,9 @@ from transformers import BertTokenizer
 from rank_bm25 import BM25Okapi
 import gradio as gr
 
+HF_TOKEN = os.getenv('HF_TOKEN')
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "budu_search_data")
+
 tokenizer = BertTokenizer.from_pretrained("DeepPavlov/rubert-base-cased")
 
 f = open('budu_search_syn_database.json')
@@ -32,7 +35,7 @@ def predict_bm25(service):
 demo = gr.Interface(fn=predict_bm25,inputs=gr.components.Textbox(label='Запрос пользователя'),
                     outputs=[gr.components.Textbox(label='Рекомендованные услуги')],
                     allow_flagging='manual',
-                    flagging_callback = gr.CSVLogger(),
+                    flagging_callback = hf_writer,
                     examples=[
                         ['кальций'],
                         ['узи'],