Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from keras.datasets import mnist
 from keras.models import Sequential,Model
+import matplotlib.pyplot as plt
 from keras.layers import Conv2D,MaxPooling2D,AveragePooling2D,InputLayer,Dense,Flatten
 option = st.sidebar.selectbox("Datasets",["Select dataset","Hand Writen Digit Dataset"])
 if option == "Hand Writen Digit Dataset":