|
--- |
|
tags: |
|
- autotrain |
|
- translation |
|
language: |
|
- fr |
|
- en |
|
datasets: |
|
- ybanas/autotrain-data-fr-en-translate |
|
co2_eq_emissions: |
|
emissions: 86.90578464498235 |
|
--- |
|
# French to English Text Translation with Transformers |
|
|
|
This code allows you to translate French text into English using the `ybanas/autotrain-fr-en-translate-51410121895` model from the Transformers library. To use this code, follow the steps below: |
|
|
|
```python |
|
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM |
|
|
|
# Load the tokenizer and the model |
|
tokenizer = AutoTokenizer.from_pretrained("ybanas/autotrain-fr-en-translate-51410121895") |
|
model = AutoModelForSeq2SeqLM.from_pretrained("ybanas/autotrain-fr-en-translate-51410121895") |
|
|
|
def translate_text(french_text: str) -> str: |
|
""" |
|
Translate French text to English using the ybanas/autotrain-fr-en-translate-51410121895 model. |
|
|
|
Args: |
|
french_text (str): French text to translate. |
|
|
|
Returns: |
|
str: Translated English text. |
|
""" |
|
# Tokenize the French text |
|
inputs = tokenizer(french_text, return_tensors="pt", padding=True, truncation=True) |
|
|
|
# Generate the English translation |
|
outputs = model.generate(**inputs) |
|
|
|
# Decode the English translation |
|
english_text = tokenizer.decode(outputs[0], skip_special_tokens=True) |
|
|
|
return english_text |
|
|
|
if __name__ == "__main__": |
|
french_text = "Les enfants aiment profiter des beaux jours" |
|
english_text = translate_text(french_text) |
|
print("French text:", french_text) |
|
print("Translated English text:", english_text) |
|
``` |
|
|
|
## Usage |
|
|
|
1. Install the Transformers library by running `pip install transformers`. |
|
2. Copy the code above into a `.py` file, for example `translation.py`. |
|
3. Replace the value of the `french_text` variable with the French text you want to translate. |
|
4. Run the script with `python translation.py`. The translated English text will be displayed on the screen. |
|
|
|
This script uses the `ybanas/autotrain-fr-en-translate-51410121895` model to translate French text into English. The model is loaded using the `AutoTokenizer` and `AutoModelForSeq2SeqLM` classes from the Transformers library. The `translate_text` function takes a French text as input and returns its translation in English. |
|
|
|
# Model Trained Using AutoTrain |
|
|
|
- Problem type: Translation |
|
- Model ID: 51410121895 |
|
- CO2 Emissions (in grams): 86.9058 |
|
|
|
## Validation Metrics |
|
|
|
- Loss: 1.455 |
|
- SacreBLEU: 15.999 |
|
- Gen len: 15.299 |