Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from keras.datasets import mnist
 from keras.models import Sequential,Model
-from keras.layers import Conv2D,MaxPooling2D,AveragePooling2D,InputLayer,Dense
+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":
     (x_train,y_train),(x_test,y_test) = mnist.load_data()