modelo_huggingface.py

from transformers import pipeline
import os
from dotenv import load_dotenv

# Carregar variáveis de ambiente
load_dotenv()
token = os.getenv("HF_TOKEN")

# Função para carregar o modelo
modelo = pipeline("text2text-generation", model="unicamp-dl/ptt5-base-portuguese-vocab", use_auth_token=token)

# Função para gerar resposta usando o modelo Hugging Face
def modelo_resposta(prompt):
    resposta = modelo(prompt)
    return resposta[0]['generated_text']