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) |
 
 ---
 
@@ -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/BART-IT-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)