| import streamlit as st |
| import numpy as np |
| import tensorflow as tf |
| from PIL import Image, ImageOps |
|
|
| model = tf.keras.models.load_model("model.keras", compile=False) |
|
|
| st.title(" Handwritten Digit Detection") |
| st.write("Upload an image of a digit (28x28 grayscale preferred).") |
|
|
| uploaded_file = st.file_uploader("Choose an image...", type=["png", "jpg", "jpeg"]) |
|
|
| if uploaded_file is not None: |
| image = Image.open(uploaded_file).convert('L') |
| image = ImageOps.invert(image) |
| image = image.resize((28, 28)) |
| img_array = np.array(image).reshape(1, 28, 28, 1) / 255.0 |
|
|
| st.image(image, caption="Processed Input", width=150) |
| pred = model.predict(img_array) |
| st.write(f"### Predicted Digit: `{np.argmax(pred)}`") |
|
|
