fix

app.py

@@ -13,23 +13,23 @@ from annotated_text import annotated_text
 ABSOLUTE_PATH = os.path.dirname(__file__)
 ASSETS_PATH = os.path.join(ABSOLUTE_PATH, 'model_assets')
 
-@st.cache(suppress_st_warning=True)
+@st.cache(suppress_st_warning=True, allow_output_mutation=True)
 def preprocess_text(s):
  return list(filter(lambda x: x!= '', (''.join(c if c.isalnum() or c == ' ' else ' ' for c in s)).split(' ')))
 
-@st.cache(suppress_st_warning=True)
+@st.cache(suppress_st_warning=True, allow_output_mutation=True)
 def get_pairwise_distances(model):
 	df = pd.read_csv(f"{ASSETS_PATH}/{model}/pairwise_distances.csv").set_index('index')
 	return df
 
-@st.cache(suppress_st_warning=True)
+@st.cache(suppress_st_warning=True, allow_output_mutation=True)
 def get_pairwise_distances_chunked(model, chunk):
 	# for df in pd.read_csv(f"{ASSETS_PATH}/{model}/pairwise_distances.csv", chunksize = 16):
 	# 	print(df.iloc[0]['queries'])
 	# 	if chunk == int(df.iloc[0]['queries']):
 	# 		return df
 	return get_pairwise_distances(model)
-@st.cache(suppress_st_warning=True)
+@st.cache(suppress_st_warning=True, allow_output_mutation=True)
 def get_query_strings():
 	df = pd.read_json(f"{ASSETS_PATH}/IUR_Reddit_test_queries_english.jsonl", lines = True)
 	df['index'] = df.reset_index().index
@@ -38,7 +38,7 @@ def get_query_strings():
 	# df.to_parquet(f"{ASSETS_PATH}/IUR_Reddit_test_queries_english.parquet", index = 'index', partition_cols = 'partition')
 	
 	# return pd.read_parquet(f"{ASSETS_PATH}/IUR_Reddit_test_queries_english.parquet", columns=['fullText', 'index', 'authorIDs'])
-@st.cache(suppress_st_warning=True)
+@st.cache(suppress_st_warning=True, allow_output_mutation=True)
 def get_candidate_strings():
 	df = pd.read_json(f"{ASSETS_PATH}/IUR_Reddit_test_candidates_english.jsonl", lines = True)
 	df['i'] = df['index']
@@ -49,24 +49,24 @@ def get_candidate_strings():
 	# df['partition'] = df['index']%100
 	# df.to_parquet(f"{ASSETS_PATH}/IUR_Reddit_test_candidates_english.parquet", index = 'index', partition_cols = 'partition')
 	# return pd.read_parquet(f"{ASSETS_PATH}/IUR_Reddit_test_candidates_english.parquet", columns=['fullText', 'index', 'authorIDs'])
-@st.cache(suppress_st_warning=True)
+@st.cache(suppress_st_warning=True, allow_output_mutation=True)
 def get_embedding_dataset(model):
 	data = load_from_disk(f"{ASSETS_PATH}/{model}/embedding")
 	return data
-@st.cache(suppress_st_warning=True)
+@st.cache(suppress_st_warning=True, allow_output_mutation=True)
 def get_bad_queries(model):
 	df = get_query_strings().iloc[list(get_pairwise_distances(model)['queries'].unique())][['fullText', 'index', 'authorIDs']]
 	return df
-@st.cache(suppress_st_warning=True)
+@st.cache(suppress_st_warning=True, allow_output_mutation=True)
 def get_gt_candidates(model, author):
 	gt_candidates = get_candidate_strings()
 	df = gt_candidates[gt_candidates['authorIDs'] == author]
 	return df
-@st.cache(suppress_st_warning=True)
+@st.cache(suppress_st_warning=True, allow_output_mutation=True)
 def get_candidate_text(l):
 	return get_candidate_strings().at[l,'fullText']
 
-@st.cache(suppress_st_warning=True)
+@st.cache(suppress_st_warning=True, allow_output_mutation=True)
 def get_annotated_text(text, word, pos):
 	print("here", word, pos)
 	start= text.index(word, pos)