Update README.md

README.md

@@ -4,8 +4,6 @@ language:
 tags:
 - text2text-generation
 - summarization
-- legal-ai
-- italian-law
 license: mit
 datasets:
 - joelniklaus/Multi_Legal_Pile
@@ -28,6 +26,8 @@ They build upon **BART-IT** ([`morenolq/bart-it`](https://huggingface.co/morenol
   - **Trained on legal documents** such as **statutes, case law, and contracts** 📑
   - **Not fine-tuned for specific tasks** (requires further adaptation)
 
+⚠️ This specific model is pre-trained on general-purpose Italian text! Please select the best model from the table below.
+
 ## 📂 Available Models
 
 | Model | Description | Link |
@@ -38,8 +38,8 @@ They build upon **BART-IT** ([`morenolq/bart-it`](https://huggingface.co/morenol
 | **LEGIT-SCRATCH-BART** | Trained from scratch on **Italian legal texts** | [🔗 Link](https://huggingface.co/morenolq/LEGIT-SCRATCH-BART) |
 | **LEGIT-SCRATCH-BART-LSG-4096** | Trained from scratch with **LSG attention**, supporting **4,096 tokens** | [🔗 Link](https://huggingface.co/morenolq/LEGIT-SCRATCH-BART-LSG-4096) |
 | **LEGIT-SCRATCH-BART-LSG-16384** | Trained from scratch with **LSG attention**, supporting **16,384 tokens** | [🔗 Link](https://huggingface.co/morenolq/LEGIT-SCRATCH-BART-LSG-16384) |
-| **BART-IT-LSG-4096** | `morenolq/bart-it` with **LSG attention**, supporting **4,096 tokens** (no legal adaptation) | [🔗 Link](https://huggingface.co/morenolq/BART-IT-LSG-4096) 
-| **BART-IT-LSG-16384** | `morenolq/bart-it` with **LSG attention**, supporting **16,384 tokens** (no legal adaptation) | [🔗 Link](https://huggingface.co/morenolq/BART-IT-LSG-16384) |
+| **BART-IT-LSG-4096** | `morenolq/bart-it` with **LSG attention**, supporting **4,096 tokens** (⚠️ no legal adaptation) | [🔗 Link](https://huggingface.co/morenolq/BART-IT-LSG-4096) 
+| **BART-IT-LSG-16384** | `morenolq/bart-it` with **LSG attention**, supporting **16,384 tokens** (⚠️ no legal adaptation) | [🔗 Link](https://huggingface.co/morenolq/BART-IT-LSG-16384) |
 
 ---
 
@@ -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)
 
-# 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)
+# 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)
 ```
 
 ---