Update app.py

app.py

@@ -11,7 +11,7 @@ import matplotlib as mpl
 
 import numpy as np
 import streamlit as st
-import plotly.express as px
+import plotly.io as pio
 
 # For measuring the inference time.
 import time
@@ -34,7 +34,7 @@ def main():
     if st.button("Prediction"):
         # plot pca cluster plot 
         plot = plot_model(kmeans, plot = 'cluster', save = True)
-        fig = px.from_json(plot)
+        fig = pio.from_json(plot)
         st.plotly_chart(fig)
         
 if __name__ == '__main__':