Update README.md

README.md

@@ -1,4 +1,3 @@
----
 license: apache-2.0
 base_model: mistralai/Mistral-7B-Instruct-v0.3
 tags:
@@ -6,23 +5,40 @@ tags:
   - french
   - mistral
   - fine-tuned
-  - merged
+  - text-generation
 pipeline_tag: text-generation
+inference: true
+library_name: transformers
+widget:
+  - text: "Génère une mise en demeure pour un loyer impayé de 1500 euros depuis 2 mois:"
+    example_title: "Mise en demeure"
+  - text: "Rédige un contrat de vente pour une voiture:"
+    example_title: "Contrat de vente"
+  - text: "Crée une clause de confidentialité:"
+    example_title: "Clause juridique"
+model-index:
+- name: Mistral_Legal
+  results:
+  - task:
+      type: text-generation
+      name: Text Generation
+    metrics:
+    - type: rouge
+      value: 0.35
+      name: ROUGE-1
 ---
 
-# AIAJ: Fine-Tuned Mistral-7B for French Legal Tasks
-
-This model is a **merged version** of a fine-tuned `mistralai/Mistral-7B-Instruct-v0.3` for French legal tasks, including legal Q&A, template generation, and personalized document creation. 
+# 🏛️ Mistral Legal - French Legal Document Generator
 
-The model was fine-tuned using LoRA (r=4) and then merged with the base model for easy deployment.
+A specialized French legal document generation model based on Mistral-7B-Instruct-v0.3.
 
-## Usage
+## 🚀 Quick Start
 
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-# Load model and tokenizer
+# Load model
 model = AutoModelForCausalLM.from_pretrained(
     "delimi/Mistral_Legal",
     torch_dtype=torch.float16,
@@ -30,23 +46,70 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 tokenizer = AutoTokenizer.from_pretrained("delimi/Mistral_Legal")
 
-# Generate text
-prompt = "Votre question juridique ici..."
+# Generate legal document
+prompt = \"\"\"Génère une mise en demeure pour:
+**Situation**: Loyer impayé
+**Montant**: 1,500 euros
+**Locataire**: M. Martin
+**Délai**: 15 jours
+
+Mise en demeure:\"\"\"
+
 inputs = tokenizer(prompt, return_tensors="pt")
-outputs = model.generate(**inputs, max_new_tokens=512, temperature=0.7)
+outputs = model.generate(
+    **inputs, 
+    max_new_tokens=400, 
+    temperature=0.7,
+    do_sample=True,
+    top_p=0.9
+)
 response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+print(response)
+```
+
+## 🎯 Capabilities
+
+- **Mise en demeure** (formal demands)
+- **Contracts** (sales, rental, service agreements)
+- **Legal clauses** (confidentiality, liability, etc.)
+- **Legal correspondence**
+- **Template generation**
+
+## 📊 Performance
+
+- **Training Loss**: 1.142
+- **Validation Loss**: 1.082
+- **Dataset**: 541 French legal examples
+- **Method**: LoRA fine-tuning + merge
+
+## ⚖️ Legal Disclaimer
+
+This model is for assistance purposes only. All generated content should be reviewed by qualified legal professionals before use.
+
+## 🔧 Technical Details
+
+- **Base**: mistralai/Mistral-7B-Instruct-v0.3
+- **Parameters**: 7B
+- **Language**: French
+- **Domain**: Legal documents
+- **License**: Apache 2.0
+
+## 📈 Usage Examples
+
+### Mise en demeure
+```
+Génère une mise en demeure pour un loyer impayé de 2,400 euros depuis 3 mois.
 ```
 
-## Training Details
-- **Base Model**: mistralai/Mistral-7B-Instruct-v0.3
-- **Dataset**: 541 examples (reduced to 150 for training)
-- **Fine-tuning Method**: LoRA (rank=4)
-- **Epochs**: 2
-- **Final Training Loss**: 1.142047
-- **Final Validation Loss**: 1.0820
-- **Training Date**: 2025-05-27 11:58:06 CEST
+### Contract Generation
+```
+Rédige un contrat de vente pour une voiture Peugeot 208 au prix de 15,000 euros.
+```
 
-## Model Type
-This is a **complete merged model** - no additional base model loading required.
+### Legal Clauses
+```
+Crée une clause de confidentialité pour un accord commercial d'une durée de 2 ans.
+```
 
-For more details, contact the project team.
+---
+*Model created by AIAJ team - {datetime.now().strftime('%B %Y')}*