Update app.py

app.py

@@ -70,15 +70,18 @@ if st.button("🔮 開始預測"):
 
     st.subheader("🧠 模型決策解釋圖（SHAP Waterfall plot）")
 
+    # 計算 SHAP 值
     shap_values = explainer(input_df)
-    fig = shap.plots.waterfall(shap_values[0], show=False)
 
-# 強制將所有 label 替換字型
-for text in fig.axes[0].texts:
-    text.set_fontname("Noto Sans CJK TC")
-    # 修正負號顯示問題
-    if text.get_text().startswith('\u2212'):
-        text.set_text(text.get_text().replace('\u2212', '-'))
+    # 取得 Axes（shap.plots.waterfall 回傳的是 Axes）
+    ax = shap.plots.waterfall(shap_values[0], show=False)
 
-    st.pyplot(fig)
-    plt.close(fig)
+    # 設定中文字型
+    for text in ax.texts:
+        text.set_fontname("Noto Sans CJK TC")
+        if text.get_text().startswith('\u2212'):
+            text.set_text(text.get_text().replace('\u2212', '-'))
+
+    # 顯示圖表
+    st.pyplot(ax.figure)
+    plt.close(ax.figure)