Update app.py

app.py

@@ -11,7 +11,7 @@ import matplotlib as mpl
 
 import numpy as np
 import streamlit as st
-import plotly.io as pio
+import plotly.graph_objs as go
 
 # For measuring the inference time.
 import time
@@ -34,7 +34,12 @@ def main():
     if st.button("Prediction"):
         # plot pca cluster plot 
         plot = plot_model(kmeans, plot = 'cluster', save = True)
-        st.plotly_chart(plot)
+        
+        # Convert the plot to a compatible format for Streamlit
+        compatible_plot = go.Figure(data=plot['data'], layout=plot['layout'])
+    
+        # Display the plot in Streamlit
+        st.plotly_chart(compatible_plot)
         
 if __name__ == '__main__':
     main()