Add application file

app.py

@@ -51,12 +51,12 @@ def classify_output(item):
         if item['score'] >= 0.5:
             # Append the category and score to the classifications list
             classifications.append((label_dict[item['label']], item['score']))
-
+            
     # Construct and print the classification message
     if classifications:
-        classification_str = ', '.join([f"'{label}' ({score:.2f})" for label, score in classifications])
+        classification_str = ', '.join([f"{label} ({score:.2f})" for label, score in classifications])
 
-        output1 = f"The item you provided is classified as: {classification_str}"
+        output1 = classification_str
         return output1
     else:
         output2 = f"No classifications with a score of 0.5 or higher were found. \n     However, the highest probability was for: '{label_dict[highest_score_item['label']]}' ({round(item['score'],2)}) \n     Use this classification with caution due to uncertainty"
@@ -66,7 +66,13 @@ def classify_output(item):
     ######## RUN GRADIO APP #################
     #########################################
 
-demo = gr.Interface(fn=classify_output, inputs="text", outputs="text")
-demo.launch()
+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')
+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')
+demo.launch(share=True)