Update README.md

README.md

@@ -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=512, 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)
 ```
 
 ---