fixed split

app.py

@@ -9,10 +9,11 @@ from transformers import AutoModel, AutoTokenizer
 
 from datasets import load_dataset
 
-@st.cache
+#@st.cache
+@st.cache_data
 def read_data(dataset_repo='dhmeltzer/asks_validation_embedded'):
     """Read the data from huggingface."""
-    return load_dataset(dataset_repo)
+    return load_dataset(dataset_repo)['validation_asks']
 
 #@st.cache(allow_output_mutation=True)
 #def load_bert_model(name="nli-distilbert-base"):
@@ -23,7 +24,8 @@ def read_data(dataset_repo='dhmeltzer/asks_validation_embedded'):
 #def load_tokenizer(name="nli-distilbert-base"):
 #    return AutoTokenizer.from_pretrained(f'sentence-transformers/{name}')
 
-@st.cache(allow_output_mutation=True)
+#@st.cache(allow_output_mutation=True)
+@st.cache_data
 def load_faiss_index(path_to_faiss="./faiss_index_small.pickle"):
     """Load and deserialize the Faiss index."""
     with open(path_to_faiss, "rb") as h:
@@ -37,8 +39,6 @@ def main():
     #tok = load_tokenizer()
     faiss_index = load_faiss_index()
 
-    import requests
-
     model_id="sentence-transformers/nli-distilbert-base"
     
     api_url = f"https://api-inference.huggingface.co/pipeline/feature-extraction/{model_id}"