Reinserted dataset context

app.py

@@ -1,13 +1,14 @@
 import streamlit as st
 from transformers import pipeline
-#from datasets import load_dataset
+from datasets import load_dataset
 
 pipe = pipeline("question-answering")
 #pipe = pipeline("text-generation")
 
 st.title("Adrega AI Help")
-#dataset = load_dataset("andreska/adregadocs", split="test")
-context = "Adrega is a very cool company, that implements AI. Rett fra Rio is a company that specializes in body waxing and is owned by Cintia" #dataset[0]["text"]
+dataset = load_dataset("andreska/adregadocs", split="test")
+context = dataset[0]["text"]
+#context = "Adrega is a very cool company, that implements AI. Rett fra Rio is a company that specializes in body waxing and is owned by Cintia" #dataset[0]["text"]
 
 user_input = st.text_input('Ask me a question')
 if st.button("Submit"):