Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- 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
|
12 |
|
13 |
-
#
|
14 |
-
@st.cache_data
|
15 |
def load_hf_dataset():
|
16 |
"""
|
17 |
-
从Hugging Face Hub
|
18 |
这将返回一个字典列表,每个字典代表一个病人数据。
|
19 |
"""
|
20 |
try:
|
21 |
# 加载'train'分割部分的数据
|
22 |
-
|
|
|
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 |
|