Traducere automată cu un model PyTorch pre-antrenat
Echipa ta de la PyBooks lucrează la un proiect de IA care implică traducerea dintr-o limbă în alta. Aceștia doresc să folosească modele pre-antrenate pentru această sarcină, ceea ce poate economisi mult timp și resurse de antrenament. Scopul acestui exercițiu este să configurezi un model de traducere din biblioteca Transformers de la HuggingFace — mai exact modelul T5 (Text-To-Text Transfer Transformer) — și să îl folosești pentru a traduce o frază din engleză în franceză.
T5Tokenizer și T5ForConditionalGeneration au fost deja importate pentru tine.
Acest exercițiu face parte din cursul
Deep Learning pentru text cu PyTorch
Instrucțiuni pentru exercițiu
- Inițializează
tokenizerșimodelfolosind modelul pre-antrenat"t5-small". - Codifică textul de intrare cu ajutorul tokenizer-ului, asigurându-te că returnezi tensori PyTorch.
- Traduce textul de intrare folosind
modelși generează rezultatul tradus.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# Initalize tokenizer and model
tokenizer = ____.from_pretrained("t5-small")
model = ____.from_pretrained("t5-small")
input_prompt = "translate English to French: 'Hello, how are you?'"
# Encode the input prompt using the tokenizer
input_ids = ____.____(input_prompt, return_tensors="____")
# Generate the translated ouput
output = model.____(input_ids, max_length=50)
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
print("Generated text:",generated_text)