Update app.py

added flagging

app.py

@@ -2,6 +2,18 @@ import gradio as gr
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
 import pickle
+from dotenv import load_dotenv
+import os 
+
+# Load environment variables from .env file
+load_dotenv()
+HF_TOKEN = os.getenv("HF_TOKEN")  
+
+# Name of dataset to save flagged data to
+HF_dataset = "peterkros/COFOG-feedback"  # <-- Replace with your dataset repo ID
+
+# Load the HuggingFaceDatasetSaver logger
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, HF_dataset)
 
 level1_to_level2_mapping = {
     "General public services": [
@@ -233,7 +245,10 @@ iface = gr.Interface(
     title="COFOG AutoClassification",
     description=markdown_text,
     article=html_table,
-    allow_flagging="auto"  # Enables flagging
+    allow_flagging="manual",  # Enables flagging
+    flagging_options=["Incorect Level1", "Incorect Level2", "Other"],
+    flagging_callback=hf_writer,
+    
 )
 
 # Run the interface