Upload categorical_classification.py

categorical_classification.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+"""Categorical Classification.ipynb
+
+Automatically generated by Colaboratory.
+
+Original file is located at
+    https://colab.research.google.com/drive/1_5FDUSnfrkNJilGyx-hoov0pnOF1CHWn
+"""
+
+!pip install gradio transformers
+
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
+tokenizer = AutoTokenizer.from_pretrained("Softechlb/articles_classification")
+model = AutoModelForSequenceClassification.from_pretrained("Softechlb/articles_classification")
+
+from transformers import pipeline
+
+classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
+def classify_text(text):
+  result = classifier(text)[0]
+  label = result["label"]
+  return label
+
+import gradio as gr
+
+interface = gr.Interface(
+  fn=classify_text,
+  inputs="textbox",
+  outputs="text",
+  title="Categorical Classification",
+  description="Text classification model Softechlb/articles_classification from Hugging Face that can classify texts into categories.",
+)
+
+interface.launch(share=True)