inital commit

README.md

@@ -1,8 +1,8 @@
 ---
 title: Bad Stuff
-emoji: 🐠
-colorFrom: pink
-colorTo: yellow
+emoji: 🚭
+colorFrom: gray
+colorTo: gray
 sdk: gradio
 sdk_version: 4.19.1
 app_file: app.py

app.py

@@ -0,0 +1,37 @@
+import gradio as gr
+from transformers import pipeline
+
+# Load the zero-shot classification pipeline
+classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+
+def classify_text(line_item, classes):
+    # Split the classes string into a list
+    class_list = classes.split(',')
+    # Perform classification
+    results = classifier(line_item, class_list, multi_class=True)
+    
+    # Prepare the output as a dictionary {label: score}
+    output = {label: round(score, 4) for label, score in zip(results['labels'], results['scores'])}
+    
+    return output
+
+# Define Gradio interface with example
+interface = gr.Interface(
+    classify_text,
+    [
+        gr.Textbox(lines=2, placeholder="Enter Line Item Here...", label="Line Item"),
+        gr.Textbox(placeholder="Enter Classes Here, Separated by Commas", label="Classes")
+    ],
+    gr.Label(num_top_classes=None, label="Class Probability Scores"),
+    title="Bad Stuff, But So Good.",
+    description="A zero-shot classification app using facebook/bart-large-mnli model to classify text into given categories.",
+    examples=[
+        ["wijn glas x3 $18", "tobacco,alcohol"],  # Example input
+        ["Stoofvlees $25", "tobacco,alcohol"],
+        ["Marlboro 10$", "tobacco,alcohol"],
+    ]
+)
+
+# Launch the application
+if __name__ == "__main__":
+    interface.launch()

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+gradio
+torch