Spaces:
Configuration error
Configuration error
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +13 -1
src/streamlit_app.py
CHANGED
@@ -1,7 +1,19 @@
|
|
1 |
# ─────────────────────────────────────────────────────────────────────────────
|
2 |
# src/streamlit_app.py
|
3 |
# ─────────────────────────────────────────────────────────────────────────────
|
4 |
-
import os
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
os.environ["STREAMLIT_HOME"] = "/tmp/.streamlit"
|
6 |
# 關閉 CORS、關掉使用者統計(可選)
|
7 |
os.environ["STREAMLIT_SERVER_ENABLE_CORS"] = "false"
|
|
|
1 |
# ─────────────────────────────────────────────────────────────────────────────
|
2 |
# src/streamlit_app.py
|
3 |
# ─────────────────────────────────────────────────────────────────────────────
|
4 |
+
import os, joblib
|
5 |
+
from train_voice_model import train_voice_model
|
6 |
+
|
7 |
+
MODEL_PATH = os.path.join(os.getcwd(), "src", "voice_model.joblib")
|
8 |
+
if not os.path.exists(MODEL_PATH):
|
9 |
+
# 第一次启动时训练并保存
|
10 |
+
clf = train_voice_model()
|
11 |
+
os.makedirs(os.path.dirname(MODEL_PATH), exist_ok=True)
|
12 |
+
joblib.dump(clf, MODEL_PATH)
|
13 |
+
|
14 |
+
# 然后再加载
|
15 |
+
voice_clf = joblib.load(MODEL_PATH)
|
16 |
+
|
17 |
os.environ["STREAMLIT_HOME"] = "/tmp/.streamlit"
|
18 |
# 關閉 CORS、關掉使用者統計(可選)
|
19 |
os.environ["STREAMLIT_SERVER_ENABLE_CORS"] = "false"
|