WingNeville's picture
Update app.py
5f3ed4e verified
from transformers import pipeline
# Predefined context
context = """
The Scrum framework consists of Scrum Teams and their roles, events, and artifacts.
Scrum Teams are self-managing and cross-functional. The main roles include the Scrum Master,
Product Owner, and Development Team. The Daily Stand-up is a short meeting for the Scrum Team
to synchronize activities and create a plan for the next 24 hours.
"""
# Load the question-answering model
question_answerer = pipeline("question-answering", model='distilbert-base-cased-distilled-squad')
def answer_question(question):
"""Answer the question based on the predefined context."""
result = question_answerer(question=question, context=context)
return result['answer']
def main():
print("Welcome to the Simple Q&A Agent!")
print("Type 'exit' to quit.\n")
while True:
user_input = input("Ask a question: ")
if user_input.lower() == 'exit':
print("Goodbye!")
break
answer = answer_question(user_input)
print(f"Answer: {answer}\n")
if __name__ == "__main__":
main()