Update Home.py

Home.py

@@ -4,10 +4,9 @@ from keras.models import load_model
 import numpy as np
 import cv2
 
-# --- Page Config ---
+
 st.set_page_config(page_title="Digit AI", layout="centered")
 
-# --- Custom UI Styling ---
 st.markdown("""
 <style>
 @import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@500&family=Outfit:wght@300;400;600&display=swap');
@@ -65,11 +64,11 @@ p {
 </style>
 """, unsafe_allow_html=True)
 
-# --- Title + Description ---
+
 st.markdown("<h1>Digit Recognizer</h1>", unsafe_allow_html=True)
 st.markdown("<p>Draw a digit (0–9) below and see what the AI thinks it is!</p>", unsafe_allow_html=True)
 
-# --- Sidebar Settings ---
+
 st.sidebar.markdown("### ✏️ Drawing Settings")
 drawing_mode = st.sidebar.selectbox("Tool", ("freedraw", "line", "rect", "circle", "transform"))
 stroke_width = st.sidebar.slider("Stroke Width", 1, 25, 10)
@@ -77,14 +76,14 @@ stroke_color = st.sidebar.color_picker("Stroke Color", "#FFFFFF")
 bg_color = st.sidebar.color_picker("Background Color", "#000000")
 realtime_update = st.sidebar.checkbox("Update Realtime", True)
 
-# --- Load Model ---
+
 @st.cache_resource
 def load_mnist_model():
     return load_model("digit_recognization.keras")
 
 model = load_mnist_model()
 
-# --- Canvas Drawing Area ---
+
 st.markdown('<div class="canvas-wrapper">', unsafe_allow_html=True)
 canvas_result = st_canvas(
     fill_color="rgba(255, 255, 255, 0.05)",
@@ -99,7 +98,7 @@ canvas_result = st_canvas(
 )
 st.markdown('</div>', unsafe_allow_html=True)
 
-# --- Prediction Display ---
+
 if canvas_result.image_data is not None:
     img = cv2.cvtColor(canvas_result.image_data.astype("uint8"), cv2.COLOR_RGBA2GRAY)
     img_resized = cv2.resize(img, (28, 28))