Update app.py

app.py

@@ -1,3 +1,15 @@
+import streamlit as st
+import numpy as np
+from PIL import Image
+from tensorflow.keras.models import load_model
+
+# Load trained model (placed in same directory as app.py)
+@st.cache_resource  # cache so model loads only once
+def load_cnn_model():
+    return load_model("mnist_cnn.h5")
+
+model = load_cnn_model()
+
 # ---- PAGE CONFIG ----
 st.set_page_config(
     page_title="Digit Recognition App",