from transformers import pipeline
from config import MODEL_NAME

def load_model():
    return pipeline("text2text-generation", model=MODEL_NAME)

def generate_answer(pipe, question):
    result = pipe(question, max_length=300, do_sample=True, top_p=0.9, temperature=0.7)
    return result[0]["generated_text"]