Update app

app

@@ -1,4 +1,7 @@
-# Sample context and question
+from transformers import pipeline
+
+qa_pipeline = pipeline("question-answering")
+
 context = """Saudi Arabia, officially known as the Kingdom of Saudi Arabia, is the largest country in the Middle East.
 It is bordered by Jordan, Iraq, Kuwait, Bahrain, Qatar, the United Arab Emirates, Oman, and Yemen.
 The capital city is Riyadh. Saudi Arabia is known for its vast deserts, including the Rub’ al Khali, also 
@@ -9,19 +12,7 @@ is Arabic, and the currency is the Saudi riyal."""
 
 question = "What is the capital of Saudi Arabia?"
 
-# Function to extract the answer from the context
-def find_answer(context, question):
-    if "capital" in question.lower():
-        # Searching for the word "capital" in the context
-        sentences = context.split(".")
-        for sentence in sentences:
-            if "capital" in sentence.lower():
-                # Assuming the answer follows after the word "capital"
-                words = sentence.split()
-                if "capital" in words:
-                    index = words.index("capital")
-                    # Return the next word which should be the capital city
-                    return words[index + 3]  # The word after 'capital city is'
+answer = qa_pipeline(question=question, context=context)
+
+print(f"Answer: {answer['answer']}")
 
-answer = find_answer(context, question)
-print(f"Answer: {answer}")