Adding Flagging Features on Gradio 3.17.0

app.py

@@ -122,6 +122,9 @@ examples = [
     ["Pemandangan di pantai kuta Bali sangatlah indah.", 30, 50, 0.5, 40, 0.98, 10, 1.0, True],
     ["Pemandangan di pantai kuta Bali sangatlah indah.", 10, 30, 1.5, 30, 0.93, 5, 2.0, True]]
 
+HF_TOKEN = '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'
+callback = gr.HuggingFaceDatasetSaver(HF_TOKEN, "output-gpt2-indo-textgen")
+
 # using gradio block to create the interface
 with gr.Blocks(title="GPT-2 Indonesian Text Generation Playground", theme='Default') as app:
     gr.Markdown("""
@@ -168,6 +171,10 @@ with gr.Blocks(title="GPT-2 Indonesian Text Generation Playground", theme='Defau
                     output1=gr.Textbox(lines=5, max_lines=50, label="Generated Text with Greedy/Beam Search Decoding")
                     button1=gr.Button("Run the model")
                     button1.click(fn=single_generation, inputs=input1, outputs=output1, show_progress=True)
+                    flag_btn = gr.Button("Flag")
+            
+            callback.setup([input1,output1],"Flagged Data Points")
+            flag_btn.click(lambda *args: callback.flag(args), input1, output1, preprocess=False)
             gr.Examples(examples, inputs=input1)
 
         #multiple generation
@@ -189,6 +196,10 @@ with gr.Blocks(title="GPT-2 Indonesian Text Generation Playground", theme='Defau
                             gr.Textbox(lines=5, max_lines=50, label="#3 Generated Text with Greedy/Beam Search Decoding")]
                     button2=gr.Button("Run the model")
                     button2.click(fn=multiple_generation, inputs=input2, outputs=output2, show_progress=True)
+                    flag_btn = gr.Button("Flag")
+
+            callback.setup([input2,output2],"Flagged Data Points")
+            flag_btn.click(lambda *args: callback.flag(args), input2, output2, preprocess=False)
             gr.Examples(examples, inputs=input2)
     
     gr.Markdown("""<p style='text-align: center'>Copyright Anugrah Akbar Praramadhan 2023 <br>