metadata
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
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))