sci-m-wang commited on
Commit
45f6809
·
verified ·
1 Parent(s): cf9b198

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +7 -6
src/streamlit_app.py CHANGED
@@ -8,18 +8,18 @@ from datasets import load_dataset # 导入Hugging Face datasets库
8
  # =======================================================================
9
 
10
  # 请将 'your_agent_file' 替换为包含 MsPatient 类的实际文件名
11
- from ms_patient import MsPatient
12
 
13
- # 使用Streamlit的缓存功能来加载和缓存数据集
14
- @st.cache_data
15
  def load_hf_dataset():
16
  """
17
- 从Hugging Face Hub加载并缓存数据集。
18
  这将返回一个字典列表,每个字典代表一个病人数据。
19
  """
20
  try:
21
  # 加载'train'分割部分的数据
22
- dataset = load_dataset("sci-m-wang/Anna-CPsyCounD", split='train')
 
23
  # 转换为pandas DataFrame再转为字典列表,方便处理
24
  return dataset.to_pandas().to_dict('records')
25
  except Exception as e:
@@ -27,6 +27,7 @@ def load_hf_dataset():
27
  return []
28
 
29
  # 加载数据
 
30
  ALL_PATIENTS = load_hf_dataset()
31
 
32
 
@@ -37,7 +38,7 @@ ALL_PATIENTS = load_hf_dataset()
37
  # --- 页面配置 ---
38
  st.set_page_config(
39
  page_title="与Anna对话",
40
- page_icon="",
41
  layout="wide"
42
  )
43
 
 
8
  # =======================================================================
9
 
10
  # 请将 'your_agent_file' 替换为包含 MsPatient 类的实际文件名
11
+ from your_agent_file import MsPatient
12
 
13
+ # 从Hugging Face加载数据集的函数(移除了缓存)
 
14
  def load_hf_dataset():
15
  """
16
+ 从Hugging Face Hub加载数据集。
17
  这将返回一个字典列表,每个字典代表一个病人数据。
18
  """
19
  try:
20
  # 加载'train'分割部分的数据
21
+ # cache_dir=None 可以尝试避免写入受限目录,但最可靠的方式是完全移除缓存
22
+ dataset = load_dataset("sci-m-wang/Anna-CPsyCounD", split='train', cache_dir=None)
23
  # 转换为pandas DataFrame再转为字典列表,方便处理
24
  return dataset.to_pandas().to_dict('records')
25
  except Exception as e:
 
27
  return []
28
 
29
  # 加载数据
30
+ # 注意:现在每次脚本重新运行时都会调用此函数
31
  ALL_PATIENTS = load_hf_dataset()
32
 
33
 
 
38
  # --- 页面配置 ---
39
  st.set_page_config(
40
  page_title="与Anna对话",
41
+ page_icon="👩",
42
  layout="wide"
43
  )
44