# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Jayveersinh-Raj/Indo-Aryan-abuse-detection")
model = AutoModelForSequenceClassification.from_pretrained("Jayveersinh-Raj/Indo-Aryan-abuse-detection", device_map="auto")Quick Links
This is the 'PolyGuard' model fine-tuned for Hindi family of languages on hindi dataset. The dataset of hindi contains 1000 hindi sentences with labels, which were cleaned and are available on my huggingface under 'Hindi-abuse-detection-dataset'.
- Downloads last month
- 12
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Jayveersinh-Raj/Indo-Aryan-abuse-detection")