Cómo ejecutar modelos de huggingface Helsinki-NLP

Answer

1. 

   davy.ai

   May 27, 2023 at 3:34 pm

   Para utilizar los modelos de Helsinki-NLP de huggingface, puedes seguir estos pasos:

   1. Instala la biblioteca transformers ejecutando pip install transformers en tu terminal.

   2. Importa los módulos necesarios y carga el modelo:

   from transformers import MarianMTModel, MarianTokenizer
   
   model_name = "Helsinki-NLP/opus-mt-es-en"
   model = MarianMTModel.from_pretrained(model_name)
   tokenizer = MarianTokenizer.from_pretrained(model_name)
   

   1. Utiliza el model y tokenizer para traducir texto:

   text = "Este es un ejemplo"
   input_ids = tokenizer(text, return_tensors="pt").input_ids
   outputs = model.generate(input_ids)
   decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)
   print(decoded)
   

   Este ejemplo traduce el texto en español “Este es un ejemplo” al inglés usando el modelo opus-mt-es-en. La salida debería ser “This is an example.”

   Puedes reemplazar el model_name con el nombre de cualquier otro modelo de Helsinki-NLP que desees usar.

Comments are closed.