Reference test document instead of main

app.py

@@ -6,7 +6,7 @@ from datasets import load_dataset
 qa_pipeline = pipeline("question-answering", model="microsoft/Phi-3-mini-4k-instruct")
 st.title("Adrega AI Help")
 dataset = load_dataset("andreska/adregadocs", split="train")
-context = dataset[0]["text"]
+context = dataset[1]["text"]