response_generator.py

from transformers import pipeline

qa_model = pipeline("question-answering", model="neuralmind/bert-base-portuguese-cased")

def generate_response(query, relevant_chunks):
    context = " ".join(relevant_chunks)
    answer = qa_model(question=query, context=context)
    return answer['answer']