Update README.md

README.md

@@ -66,7 +66,7 @@ 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/BART-IT-LSG-4096"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = BartForConditionalGeneration.from_pretrained(model_name)
 
@@ -74,10 +74,10 @@ model = BartForConditionalGeneration.from_pretrained(model_name)
 input_text = "<mask> 1234: Il contratto si intende concluso quando..."
 inputs = tokenizer(input_text, return_tensors="pt", max_length=4096, truncation=True)
 
-# Generate summary
-summary_ids = model.generate(inputs.input_ids, max_length=150, num_beams=4, early_stopping=True)
-summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
-print("📝 Summary:", summary)
+# Pre-trained model fill the mask
+output_ids = model.generate(inputs.input_ids, max_length=150, num_beams=4, early_stopping=True)
+output_text = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+print("📝:", output_text)
 ```
 
 ---