Add application file

app.py

@@ -0,0 +1,16 @@
+from transformers import AutoTokenizer
+import gradio as gr
+from transformers import AutoModelForSequenceClassification
+from transformers import pipeline
+
+id2label = {0:'INSULT', 1:'OTHER', 2:'PROFANITY', 3:'RACIST', 4:'SEXIST'}
+model = AutoModelForSequenceClassification.from_pretrained(".")
+tokenizer = AutoTokenizer.from_pretrained(".")
+
+def predict(text):
+	classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
+	result = id2label[int(classifier(text)[0]['label'].split('_')[-1])]
+	return result
+
+iface = gr.Interface(fn=predict, inputs="text", outputs="text")
+iface.launch()

requirements.txt

@@ -0,0 +1,4 @@
+torch
+torchvision
+transformers==4.18.0
+numpy<1.24.0