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,
|