tyang4 commited on
Commit
6bb5846
·
verified ·
1 Parent(s): c6d7da1

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +3 -2
src/streamlit_app.py CHANGED
@@ -550,8 +550,9 @@ def save_dataset(df: pd.DataFrame, filename: str) -> str:
550
  if len(df) < 10:
551
  st.warning(f"❌ Dataset too small to save: only {len(df)} rows.")
552
  return ""
553
- os.makedirs("saved_datasets", exist_ok=True)
554
- path = f"saved_datasets/{filename}.csv"
 
555
  if os.path.exists(path):
556
  old_hash = hashlib.md5(open(path, 'rb').read()).hexdigest()
557
  new_hash = hashlib.md5(df.to_csv(index=False).encode()).hexdigest()
 
550
  if len(df) < 10:
551
  st.warning(f"❌ Dataset too small to save: only {len(df)} rows.")
552
  return ""
553
+ save_dir = "/tmp/saved_datasets"
554
+ os.makedirs(save_dir, exist_ok=True)
555
+ path = f"{save_dir}/{filename}.csv"
556
  if os.path.exists(path):
557
  old_hash = hashlib.md5(open(path, 'rb').read()).hexdigest()
558
  new_hash = hashlib.md5(df.to_csv(index=False).encode()).hexdigest()