Upload folder using huggingface_hub

app.py

@@ -66,7 +66,7 @@ st.markdown(f"<h2 style='color:green;'>{predicted_repayment_interval}</h2>", uns
 
 # Explanation for SHAP force plots
 st.write("## 🔍 SHAP Explanation")
-st.markdown("The following SHAP plots explain the model's decision for each repayment interval type. These visualizations help you understand the key features that influenced the model's prediction.")
+st.markdown("The following SHAP plot explains the model's decision for the predicted repayment interval. This visualization helps you understand the key features that influenced the model's prediction.")
 
 # SHAP explanations
 explainer = shap.TreeExplainer(model_xgb)
@@ -77,50 +77,48 @@ def add_background(html_content):
     white_background_style = "<style>body { background-color: white; }</style>"
     return white_background_style + html_content
 
-# Generate and save SHAP force plot for Class 0 (Bullet)
-st.write("### 🚅 SHAP Force Plot for Class 0: Bullet Repayment Interval")
-st.markdown("This plot explains the factors influencing the Bullet repayment interval prediction. Bullet repayment means paying off the loan in one lump sum at the end of the loan period.")
-shap_html_path_0 = "shap_force_plot_class_0.html"
-shap.save_html(shap_html_path_0, shap.force_plot(
-    explainer.expected_value[0],
-    shap_values[0][:, 0],
-    X_processed.iloc[0, :].values,
-    feature_names,
-    show=False,
-    matplotlib=False
-))
-with open(shap_html_path_0, 'r', encoding='utf-8') as f:
-    shap_html_0 = f.read()
-st.components.v1.html(add_background(shap_html_0), height=130)
-
-# Generate and save SHAP force plot for Class 1 (Irregular)
-st.write("### 🪙 SHAP Force Plot for Class 1: Irregular Repayment Interval")
-st.markdown("This plot explains the factors influencing the Irregular repayment interval prediction. Irregular repayment means paying off the loan at irregular intervals based on specific conditions.")
-shap_html_path_1 = "shap_force_plot_class_1.html"
-shap.save_html(shap_html_path_1, shap.force_plot(
-    explainer.expected_value[1],
-    shap_values[0][:, 1],
-    X_processed.iloc[0, :].values,
-    feature_names,
-    show=False,
-    matplotlib=False
-))
-with open(shap_html_path_1, 'r', encoding='utf-8') as f:
-    shap_html_1 = f.read()
-st.components.v1.html(add_background(shap_html_1), height=130)
-
-# Generate and save SHAP force plot for Class 2 (Monthly)
-st.write("### 📅 SHAP Force Plot for Class 2: Monthly Repayment Interval")
-st.markdown("This plot explains the factors influencing the Monthly repayment interval prediction. Monthly repayment means paying off the loan in equal monthly installments.")
-shap_html_path_2 = "shap_force_plot_class_2.html"
-shap.save_html(shap_html_path_2, shap.force_plot(
-    explainer.expected_value[2],
-    shap_values[0][:, 2],
-    X_processed.iloc[0, :].values,
-    feature_names,
-    show=False,
-    matplotlib=False
-))
-with open(shap_html_path_2, 'r', encoding='utf-8') as f:
-    shap_html_2 = f.read()
-st.components.v1.html(add_background(shap_html_2), height=131)
+# Generate and display SHAP force plot based on the predicted repayment interval
+if predicted_encoded_repayment_interval == 0:
+    st.write("### 🚅 SHAP Force Plot for Bullet Repayment Interval")
+    shap_html_path_0 = "shap_force_plot_class_0.html"
+    shap.save_html(shap_html_path_0, shap.force_plot(
+        explainer.expected_value[0],
+        shap_values[0][:, 0],
+        X_processed.iloc[0, :].values,
+        feature_names,
+        show=False,
+        matplotlib=False
+    ))
+    with open(shap_html_path_0, 'r', encoding='utf-8') as f:
+        shap_html_0 = f.read()
+    st.components.v1.html(add_background(shap_html_0), height=130)
+
+elif predicted_encoded_repayment_interval == 1:
+    st.write("### 🪙 SHAP Force Plot for Irregular Repayment Interval")
+    shap_html_path_1 = "shap_force_plot_class_1.html"
+    shap.save_html(shap_html_path_1, shap.force_plot(
+        explainer.expected_value[1],
+        shap_values[0][:, 1],
+        X_processed.iloc[0, :].values,
+        feature_names,
+        show=False,
+        matplotlib=False
+    ))
+    with open(shap_html_path_1, 'r', encoding='utf-8') as f:
+        shap_html_1 = f.read()
+    st.components.v1.html(add_background(shap_html_1), height=130)
+
+elif predicted_encoded_repayment_interval == 2:
+    st.write("### 📅 SHAP Force Plot for Monthly Repayment Interval")
+    shap_html_path_2 = "shap_force_plot_class_2.html"
+    shap.save_html(shap_html_path_2, shap.force_plot(
+        explainer.expected_value[2],
+        shap_values[0][:, 2],
+        X_processed.iloc[0, :].values,
+        feature_names,
+        show=False,
+        matplotlib=False
+    ))
+    with open(shap_html_path_2, 'r', encoding='utf-8') as f:
+        shap_html_2 = f.read()
+    st.components.v1.html(add_background(shap_html_2), height=130)