# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("spuun/BERATBOS", dtype="auto")Quick Links
BERATBOS v0.0.1-prealpha
BERT-based Automated Text Classification Based on POS
A model I made for my final year assignment, used a random script to train this on GPU, as a result, the usage is kinda non-standard, as far as transformers-based models go, at least.
Model is for detecting AI-generated works via coherency comparison through POS tag differences, it was only trained for 5 epochs on a dataset of 1000 human sentences and 1000 BLOOM-generated sentences.
I have plans to further this model into an actual production model, but will only be done once I can properly assess on whether that's a worthy enough of an endeavour.
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="spuun/BERATBOS")