Update app.py

app.py

@@ -71,7 +71,7 @@ from streamlit_drawable_canvas import st_canvas
 # === Load models ===
 @st.cache_resource
 def load_single_digit_model():
-    return load_model("mnist_model.keras")
+    return load_model("single_digit_model.keras")
 
 @st.cache_resource
 def load_multi_digit_model():
@@ -133,7 +133,9 @@ if canvas_result.image_data is not None:
     # Preprocess image
     img = cv2.cvtColor(canvas_result.image_data.astype("uint8"), cv2.COLOR_RGBA2GRAY)
     img = 255 - img  # Invert colors
-    img_resized = cv2.resize(img, (28, 80))  # Resize to match multi-digit model input shape
+    
+    # Resize image to match model input dimensions
+    img_resized = cv2.resize(img, (80, 28))  # Resize to match multi-digit model input shape
     img_normalized = img_resized / 255.0
     final_img = img_normalized.reshape(1, 28, 80, 1)
 
@@ -157,4 +159,3 @@ if canvas_result.image_data is not None:
 
     # Show prediction result
     st.markdown(f"### 🧠 Predicted Number: **{predicted_str}**")
-