Views
No views yet
1from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2
3# Load the model and tokenizer
4model = AutoModelForSeq2SeqLM.from_pretrained("giuid/flan_t5_xl_summarization_v2")
5tokenizer = AutoTokenizer.from_pretrained("giuid/flan_t5_xl_summarization_v2")
6
7# Input text
8input_text = "Your lengthy legal document text here..."
9
10# Tokenize and generate summary
11inputs = tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True)
12outputs = model.generate(inputs.input_ids, max_length=150, num_beams=5, early_stopping=True)
13
14# Decode summary
15summary = tokenizer.decode(outputs[0], skip_special_tokens=True)
16print(summary)