Reverted trying to not load dataset, as it causes problem with requests

app.py

@@ -7,6 +7,18 @@ from huggingface_hub import InferenceClient
 api_key = os.getenv("HF_API_KEY")
 client = InferenceClient(api_key=api_key)
 
+# Load the dataset
+dataset = load_dataset("andreska/adregadocs", split="test")
+
+# Function to read the content from the dataset
+def read_dataset(dataset):
+    text = []
+    for item in dataset:
+        text.append(item['text'])
+    return "\n".join(text)
+
+context = read_dataset(dataset)
+
 # Inject custom CSS to change the background color to yellow
 st.markdown(
     """
@@ -27,20 +39,6 @@ if 'conversation' not in st.session_state:
 if 'include_context' not in st.session_state: 
     st.session_state.include_context = False
 
-if include_context:
-    dataset = load_dataset("andreska/adregadocs", split="test")
-
-    # Function to read the content from the dataset
-    def read_dataset(dataset):
-        text = []
-        for item in dataset:
-            text.append(item['text'])
-        return "\n".join(text)
-
-    context = read_dataset(dataset)
-else:
-    context = ""
-
 def handle_submit():
     user_input = st.session_state.user_input
     if user_input: