Update app.py

app.py

@@ -32,8 +32,10 @@ def main():
 
     if st.button("Prediction"):
         # plot pca cluster plot 
-        plot_model(kmeans, plot = 'cluster')
-
+        plot = plot_model(kmeans, plot = 'cluster', save = True)
+        fig = go.Figure()
+        fig = go.FigureWidget(plot)
+        st.plotly_chart(fig)
         
 if __name__ == '__main__':
     main()