for scum and villainy

README.md

@@ -1,5 +1,5 @@
 ---
-title: Toolbox
+title: Fandom Classifier with TextClassificationTool
 emoji: 🐨
 colorFrom: blue
 colorTo: indigo
@@ -10,4 +10,7 @@ pinned: false
 license: cc-by-sa-3.0
 ---
 
+# Fandom Classifier
+
+This is the Fandom Classifier, a demo of the transformers TextClassificationTool.
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,25 @@
+import gradio as gr
+
+from transformers.tools.text_classification import TextClassificationTool
+
+classifier = TextClassificationTool()
+labels = ["Star Wars", "Star Trek", "MCU", "LOTR", "Babylon 5", "DC Comics"]
+
+iface = gr.Interface(
+    fn=lambda s: classifier(s, labels=labels),
+    inputs="text",
+    outputs="text")
+with gr.Blocks() as app:
+    story=gr.TextArea(
+        label='Story',
+        placeholder='The space pirate drew his trusty blaster. This might be his last night in Mos Eisley')
+    output=gr.Textbox(label='Fandom')
+    note=gr.Markdown(f"I know {', '.join(labels)}")
+    button = gr.Button('Classify')
+    button.click(
+        fn=lambda s: classifier(s, labels=labels),
+        inputs=story,
+        outputs=output,
+        api_name="fandom")
+                 
+app.launch()