Update README.md

README.md

@@ -66,13 +66,13 @@ They build upon **BART-IT** ([`morenolq/bart-it`](https://huggingface.co/morenol
 from transformers import BartForConditionalGeneration, AutoTokenizer
 
 # Load tokenizer and model
-model_name = "morenolq/LEGIT-BART"
+model_name = "morenolq/LEGIT-BART-LSG-16384"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = BartForConditionalGeneration.from_pretrained(model_name)
 
 # Example input
 input_text = "<mask> 1234: Il contratto si intende concluso quando..."
-inputs = tokenizer(input_text, return_tensors="pt", max_length=4096, truncation=True)
+inputs = tokenizer(input_text, return_tensors="pt", max_length=16384, truncation=True)
 
 # Generate summary
 summary_ids = model.generate(inputs.input_ids, max_length=150, num_beams=4, early_stopping=True)