Spaces:
Running
on
Zero
Running
on
Zero
Update app_dialogue.py
Browse files- app_dialogue.py +6 -5
app_dialogue.py
CHANGED
@@ -297,7 +297,8 @@ chatbot = gr.Chatbot(
|
|
297 |
|
298 |
dope_callback = gr.CSVLogger()
|
299 |
problematic_callback = gr.CSVLogger()
|
300 |
-
|
|
|
301 |
|
302 |
# Using Flagging for saving dope and problematic examples
|
303 |
# Dope examples flagging
|
@@ -385,7 +386,7 @@ with gr.Blocks(
|
|
385 |
dope_bttn = gr.Button("Dope🔥")
|
386 |
with gr.Column(scale=1, min_width=50):
|
387 |
problematic_bttn = gr.Button("Problematic😬")
|
388 |
-
|
389 |
[
|
390 |
model_selector,
|
391 |
chatbot,
|
@@ -398,7 +399,7 @@ with gr.Blocks(
|
|
398 |
"gradio_dope_data_points",
|
399 |
)
|
400 |
dope_bttn.click(
|
401 |
-
lambda *args:
|
402 |
[
|
403 |
model_selector,
|
404 |
chatbot,
|
@@ -412,7 +413,7 @@ with gr.Blocks(
|
|
412 |
preprocess=False,
|
413 |
)
|
414 |
# Problematic examples flagging
|
415 |
-
|
416 |
[
|
417 |
model_selector,
|
418 |
chatbot,
|
@@ -425,7 +426,7 @@ with gr.Blocks(
|
|
425 |
"gradio_problematic_data_points",
|
426 |
)
|
427 |
problematic_bttn.click(
|
428 |
-
lambda *args:
|
429 |
[
|
430 |
model_selector,
|
431 |
chatbot,
|
|
|
297 |
|
298 |
dope_callback = gr.CSVLogger()
|
299 |
problematic_callback = gr.CSVLogger()
|
300 |
+
dope_dataset_writer = gr.HuggingFaceDatasetSaver(HF_WRITE_TOKEN, "HuggingFaceM4/dope-dataset")
|
301 |
+
problematic_dataset_writer = gr.HuggingFaceDatasetSaver(HF_WRITE_TOKEN, "HuggingFaceM4/problematic-dataset")
|
302 |
|
303 |
# Using Flagging for saving dope and problematic examples
|
304 |
# Dope examples flagging
|
|
|
386 |
dope_bttn = gr.Button("Dope🔥")
|
387 |
with gr.Column(scale=1, min_width=50):
|
388 |
problematic_bttn = gr.Button("Problematic😬")
|
389 |
+
dope_dataset_writer.setup(
|
390 |
[
|
391 |
model_selector,
|
392 |
chatbot,
|
|
|
399 |
"gradio_dope_data_points",
|
400 |
)
|
401 |
dope_bttn.click(
|
402 |
+
lambda *args: dope_dataset_writer.flag(args),
|
403 |
[
|
404 |
model_selector,
|
405 |
chatbot,
|
|
|
413 |
preprocess=False,
|
414 |
)
|
415 |
# Problematic examples flagging
|
416 |
+
problematic_dataset_writer.setup(
|
417 |
[
|
418 |
model_selector,
|
419 |
chatbot,
|
|
|
426 |
"gradio_problematic_data_points",
|
427 |
)
|
428 |
problematic_bttn.click(
|
429 |
+
lambda *args: problematic_dataset_writer.flag(args),
|
430 |
[
|
431 |
model_selector,
|
432 |
chatbot,
|