expand-questions

utils/loaders.py

@@ -13,7 +13,7 @@ DATA_FILES = os.getenv("GEMINI_DATA_FILES")
 @st.cache_data
 def load_data():
     try:
-        data = pd.read_csv("data/user-evaluation/merged.csv")[:5]
+        data = pd.read_csv("data/user-evaluation/merged.csv")
         return data
     except Exception as e:
 
@@ -22,7 +22,7 @@ def load_data():
         dataset = load_dataset(REPO_NAME, token=True, data_files=DATA_FILES, revision="main")
         dataset.set_format(type='pandas')  ## converting it into pandas
         df = dataset["train"][:]
-        return df[:5]
+        return df
 
 
 def load_html(file_name):