Upload 4 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+mnist_model.keras filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,12 +1,35 @@
 ---
-title: Mnist Model Joe
-emoji: 👀
-colorFrom: gray
-colorTo: pink
-sdk: gradio
-sdk_version: 6.2.0
-app_file: app.py
-pinned: false
+library_name: keras
+tags:
+- image-classification
+- tensorflow
+datasets:
+- mnist
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# MNIST Digit Classifier
+
+This is a Keras/TensorFlow model trained on the MNIST dataset to classify handwritten digits (0-9).
+
+## Model Details
+- **Framework:** TensorFlow / Keras
+- **Task:** Image Classification (Multi-class)
+- **Dataset:** MNIST
+
+## Intended Use
+This model is intended for educational purposes to demonstrate digit classification.
+
+## Usage
+
+You can download and load this model in Python:
+
+```python
+from huggingface_hub import hf_hub_download
+import tensorflow as tf
+
+# Download the model file
+model_path = hf_hub_download(repo_id="your-username/your-model-name", filename="mnist_model.keras")
+
+# Load the model
+model = tf.keras.models.load_model(model_path)
+```

app.py

@@ -0,0 +1,47 @@
+import gradio as gr
+import tensorflow as tf
+import numpy as np
+from PIL import Image, ImageOps
+
+# Load the model directly
+# reliable for Spaces where you upload the model file alongside the app
+model = tf.keras.models.load_model("mnist_model.keras")
+
+def predict_digit(image):
+    if image is None:
+        return None
+    
+    # 1. Convert to grayscale
+    image = image.convert('L')
+    
+    # 2. Resize to 28x28 to match training data
+    image = image.resize((28, 28))
+    
+    # 3. Invert colors (MNIST is white text on black background)
+    # Most user uploads are black text on white background (paper), so we usually need to invert
+    # We check mean pixel value; if > 127, it's likely a white background.
+    if np.mean(image) > 127:
+        image = ImageOps.invert(image)
+        
+    # 4. Convert to numpy array and normalize
+    image_array = np.array(image) / 255.0
+    
+    # 5. Flatten to shape (1, 784) as expected by the Dense input layer
+    image_array = image_array.reshape(1, 784)
+    
+    # Predict
+    prediction = model.predict(image_array)
+    
+    # Return dictionary for Gradio Label output
+    return {str(i): float(prediction[0][i]) for i in range(10)}
+
+iface = gr.Interface(
+    fn=predict_digit,
+    inputs=gr.Image(type="pil", label="Upload an Image"),
+    outputs=gr.Label(num_top_classes=3, label="Predictions"),
+    title="MNIST Digit Classifier",
+    description="Upload an image of a handwritten digit (0-9) to see the prediction. Works best with a single digit centered in the image."
+)
+
+if __name__ == "__main__":
+    iface.launch()

mnist_model.keras

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:24be0fec8219f8ad6247a3738889add9e4d6ac8d61b03d9a0aef362c6f4e9ea5
+size 974585

requirements.txt

@@ -0,0 +1,4 @@
+tensorflow
+gradio
+numpy
+pillow