from transformers import pipeline
task = "text-generation"
model_name = "gpt2"
max_output_length = 30
num_of_return_sequences = 2
input_text = "Hello, "
text_generator = pipeline(
task,
model = model_name)
text_generator(
input_text,
max_length=max_output_length,
num_return_sequences=num_of_return_sequences)
from transformers import pipeline
task = "text2text-generation"
model_name = "google/flan-t5-base"
input_text = "Can you convert this text to French language: Hello, How are you"
text2text_generator = pipeline(
task,
model = model_name)
text2text_generator(input_text)