You can deploy this model and start earning money today!

melanchthon19/
mews-aes-classifier-base

Name: mews-aes-classifier-base
Availability: InStock

safetensors

distilbert

automated_essay_scoring

text_classification

model_hub_mixin

pytorch_model_hub_mixin

Views

No views yet

DistilBERT Essay Scoring Model

This model is a proof-of-concept for automated essay scoring, built using DistilBERT and integrated with the Hugging Face Hub through the PyTorchModelHubMixin.
It was trained for only one epoch and is intended for testing pipeline integration, not production use.

Task

Automated Essay Scoring (Text Classification)

Usage

python

from transformers import AutoModelForSequenceClassification, AutoTokenizer
import torch

model_name = "melanchthon19/mews-aes-classifier-base"

tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)

text = "Artificial intelligence models are transforming how we write essays."
inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)

with torch.no_grad():
    outputs = model(**inputs)
    logits = outputs.logits
    probs = torch.softmax(logits, dim=-1)
    pred_id = probs.argmax(dim=-1).item()

label = model.config.id2label[pred_id]
print(f"Predicted class: {label} (index {pred_id})")
# Scores go from 1 to 5 (index 0 --> 4)