Update app.py

app.py

@@ -137,8 +137,8 @@ def main():
             components.html(HtmlFile.read(), height=600)
             st.write('True label:', labels[str(y_test[idx])])
             st.write("Effect of Predictors")
-            exp.as_pyplot_figure()
-            st.pyplot(bbox_inches='tight')
+            fig = exp.as_pyplot_figure()
+            st.pyplot(fig, bbox_inches='tight')
 
 
 if __name__ == "__main__":