| tags: | |
| - onnx | |
| - summarization | |
| - bigbird | |
| - legal | |
| library_name: transformers | |
| base_model: google/bigbird-pegasus-large-arxiv | |
| # Legal BigBird (ONNX) | |
| This is a fine-tuned version of **BigBird-Pegasus** for legal document summarization. | |
| It has been exported to **ONNX** for high-performance inference. | |
| ## Usage | |
| ```python | |
| import onnxruntime as ort | |
| from transformers import AutoTokenizer | |
| tokenizer = AutoTokenizer.from_pretrained("arirajuns/bigbird-legal-onnx") | |
| session = ort.InferenceSession("model.onnx") | |
| inputs = tokenizer("Your long legal text here...", return_tensors="np") | |
| outputs = session.run(None, dict(inputs)) | |
| ``` | |