Update app.py

app.py

@@ -33,13 +33,13 @@ def main():
 
     if st.button("Prediction"):
         # plot pca cluster plot 
-        plot = plot_model(kmeans, plot = 'cluster', save = True)
+        plot_model(kmeans, plot = 'cluster', display_format = ‘streamlit’)
         
         # Convert the plot to a compatible format for Streamlit
-        compatible_plot = go.Figure(data=plot['data'], layout=plot['layout'])
+        # compatible_plot = go.Figure(data=plot['data'], layout=plot['layout'])
     
         # Display the plot in Streamlit
-        st.plotly_chart(compatible_plot)
+        # st.plotly_chart(compatible_plot)
         
 if __name__ == '__main__':
     main()