huggingface_model.py

# huggingface_model.py
from transformers import pipeline

def load_model():
    return pipeline("text2text-generation", model="unicamp-dl/ptt5-base-portuguese-vocab")