Kom igångKom igång gratis

Språköversättning med en förtränad PyTorch-modell

Ditt team på PyBooks arbetar med ett AI-projekt som innebär översättning mellan olika språk. De vill använda förtränade modeller för uppgiften, vilket kan spara både tid och resurser. I den här övningen ska du konfigurera en översättningsmodell från HuggingFaces Transformers-bibliotek – närmare bestämt T5-modellen (Text-To-Text Transfer Transformer) – och använda den för att översätta en engelsk fras till franska.

T5Tokenizer och T5ForConditionalGeneration har redan laddats in åt dig.

Den här övningen är en del av kursen

Djupinlärning för text med PyTorch

Visa kurs

Övningsinstruktioner

  • Initiera tokenizer och model från den förtränade modellen "t5-small".
  • Koda inmatningsprompten med tokenizern och se till att returnera PyTorch-tensorer.
  • Översätt inmatningsprompten med model och generera den översatta utdatan.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

# 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)
Redigera och kör kod