Downloads · 30 days
10
4% of all-time downloads
briefme-io/legal_document_classifier
legal_document_classifier is a text classification model from briefme-io. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
This is a fine-tuned BERT-based model for classifying legal documents into contract, amendment, or other.
Downloads · 30 days
10
4% of all-time downloads
All-time downloads
257
Public
Parameters
67M
2.4 GB on disk
Likes
1
Public
Click a slice to open those files.
.safetensors268 MB · 100%
From the Hugging Face model README
This is a fine-tuned BERT-based model for classifying legal documents into contract, amendment, or other.
from transformers import AutoTokenizer, AutoModelForSequenceClassification
model = AutoModelForSequenceClassification.from_pretrained("karthikvarunn/legal_document_classifier")
tokenizer = AutoTokenizer.from_pretrained("karthikvarunn/legal_document_classifier")
inputs = tokenizer("Your document text here...", return_tensors="pt")
outputs = model(**inputs)