| pipeline_tag: text-generation | |
| library_name: transformers | |
| license: apache-2.0 | |
| language: | |
| - fr | |
| - en | |
| tags: | |
| - llama | |
| - function-calling | |
| - sncf | |
| - trains | |
| # SNCF Router Model | |
| Ce modèle est un routeur d'outils intelligent pour la recherche de trains SNCF. | |
| ## Usage | |
| ```python | |
| from transformers import AutoTokenizer, AutoModelForCausalLM | |
| tokenizer = AutoTokenizer.from_pretrained("Skapersk/sncf") | |
| model = AutoModelForCausalLM.from_pretrained("Skapersk/sncf") | |
| prompt = "J'aimerais aller à Nantes, geoloc: Paris" | |
| inputs = tokenizer(prompt, return_tensors="pt") | |
| outputs = model.generate(**inputs, max_new_tokens=200) | |
| response = tokenizer.decode(outputs[0], skip_special_tokens=True) | |
| ``` | |
| ## API d'inférence | |
| Le modèle est compatible avec l'API d'inférence Hugging Face. | |