File size: 310 Bytes
61253aa |
1 2 3 4 5 6 7 8 9 10 |
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"]
|