Add application file

app.py

@@ -8,8 +8,8 @@ import os
 #######################
 api_key = os.environ.get("HF_API_KEY_INFERENCE")
 API_URL = "https://api-inference.huggingface.co/models/chernandezc/distilbert-base-uncased-finetuned-items-multi-label-21122023-AUGMENTED" #Api endpoint.
-headers = {"Authorization": f"Bearer {api_key}"} #This is a read only API key, do not share please :D.
-                                                                            #This is visible just to experiment, I cannot share environment variables.
+headers = {"Authorization": f"Bearer {api_key}"} #Api Key. 
+                                                                            
 
 def query(payload): #Function to use the API.
 	response = requests.post(API_URL, headers=headers, json=payload)
@@ -68,11 +68,16 @@ def classify_output(item):
 
 txtbx = gr.Textbox(value = 'I would like to feel better', label = 'Please enter your item:', container = 'True')
 txtbxopt = gr.Textbox(label = 'The item you provided was classified as:', container = 'True')
+hf_writer = gr.HuggingFaceDatasetSaver(api_key, 'flagging_EEMM_V05')
 demo = gr.Interface(fn=classify_output, inputs=txtbx, outputs=txtbxopt,
                     theme = gr.themes.Soft(primary_hue='orange'),
                     title = 'EEMM Item Classification Machine V 0.5',
                     description = 'This machine is a fine tuned version of DistillBERT. It classifies items in 7 EEMM dimensions following (ref). **Please note that the machine goes idle after a period of inactivity. If this occurs, waking it up may take around 20 seconds. Be patient ;)**',
-                    article = 'Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum')
+                    article = 'Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum',
+                    allow_flagging = 'manual',
+                    flagging_options = ['Wrong category','Lacks a category'],
+                    flagging_callback= hf_writer
+)
 demo.launch()