File size: 692 Bytes
0f64169 0c9fe1b 0f64169 0c9fe1b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
---
language: en
license: mit
tags:
- text-classification
- transformers
- legal
- pdf
datasets:
- custom
model-index:
- name: legal_document_classifier
results: []
---
# Legal Document Classifier
This is a fine-tuned BERT-based model for classifying legal documents into contract, amendment, or other.
## Usage
```python
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)
|