GCLing commited on
Commit
543294f
·
verified ·
1 Parent(s): 9810119

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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"