Update app.py

app.py

@@ -113,7 +113,7 @@ def generate_adversarial_example(dataset, attacker, text=None, label=None):
         classification_df["is_repaired"] = result["is_fixed"]
         classification_df["pred_label"] = result["label"]
         classification_df["confidence"] = round(result["confidence"], 3)
-        classification_df["is_correct"] = str(result["pred_label"]) == str(label)
+        classification_df["is_correct"] = str(result["label"]) == str(label)
 
         advdetection_df = {}
         if result["is_adv_label"] != "0":
@@ -143,6 +143,7 @@ def generate_adversarial_example(dataset, attacker, text=None, label=None):
         attack_result.perturbed_result.output,
         pd.DataFrame(classification_df, index=[0]),
         pd.DataFrame(advdetection_df, index=[0]),
+        ''
     )
 
 
@@ -210,7 +211,7 @@ if __name__ == "__main__":
         with gr.Group():
             with gr.Row():
                 input_dataset = gr.Radio(
-                    choices=["SST2", "Amazon", "AGNews10K"],
+                    choices=["SST2", "Amazon", "Yahoo", "AGNews10K"],
                     value="SST2",
                     label="Select a testing dataset and an adversarial attacker to generate an adversarial example.",
                 )