Update app.py

app.py

@@ -29,13 +29,10 @@ if st.button("Explain"):
 
     # Save SHAP plot as HTML
     shap_html = shap.plots.text(shap_values, display=False)
-    
-    # Save the plot to an HTML file
-    shap_html.save_html("shap_plot.html")
-
-    # Read the HTML file and display in Streamlit
-    with open("shap_plot.html", "r") as f:
-        shap_html = f.read()
 
+    # Save the plot to an HTML file
+    with open("shap_plot.html", "w") as f:
+        f.write(shap_html)
+    
     # Display the SHAP plot in Streamlit using components
-    st.components.v1.html(shap_html, height=500, scrolling=True)
+    st.components.v1.html(shap_html, height=500, scrolling=True)