A fancy weekly newsletter generator for Wonderflow Development team. NOTE: Use with caution.
To use this model, first load it:
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("mpangrazzi/wonderflow_newsletter")
model = AutoModelForCausalLM.from_pretrained("mpangrazzi/wonderflow_newsletter")
Then, use a pipeline
to get predictions:
from transformers import pipeline
text_generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
inputs = ["This week the development team"]
samples = text_generator(
inputs,
do_sample=True,
max_length=150,
num_return_sequences=5,
num_beans=5,
top_p=0.90,
temperature=1.3
)
outputs = [entry["generated_text"] for sample in samples for entry in sample]
for entry in outputs:
print(f"{entry}\n\n")