from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

# Load model
model = AutoModelForCausalLM.from_pretrained(
    "delimi/Mistral_Legal",
    torch_dtype=torch.float16,
    device_map="auto"
)
tokenizer = AutoTokenizer.from_pretrained("delimi/Mistral_Legal")

# 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=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.

Contract Generation

Rédige un contrat de vente pour une voiture Peugeot 208 au prix de 15,000 euros.

Legal Clauses

Crée une clause de confidentialité pour un accord commercial d'une durée de 2 ans.

Model created by AIAJ team - {datetime.now().strftime('%B %Y')}