Views
No views yet
1from transformers import T5Tokenizer, T5ForConditionalGeneration
2
3# Load the model and tokenizer
4tokenizer = T5Tokenizer.from_pretrained("Fafadalilian/lora-adapter-t5_small_model_California_state_bill")
5model = T5ForConditionalGeneration.from_pretrained("Fafadalilian/lora-adapter-t5_small_model_California_state_bill")
6
7# Example input text
8input_text = "summarize: [Insert California state bill text here]"
9
10# Tokenize the input
11inputs = tokenizer(input_text, return_tensors="pt", truncation=True, padding="max_length", max_length=512)
12
13# Generate summary
14summary_ids = model.generate(inputs.input_ids, max_length=150, num_beams=2, length_penalty=2.0, early_stopping=True)
15summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
16
17print("Summary:", summary)
18