Upload 4 files

app.py

@@ -0,0 +1,35 @@
+# app.py
+import torch
+import torch.nn.functional as F
+import gradio as gr
+import numpy as np
+from PIL import Image
+from model import CNN
+
+# Load model
+model = CNN()
+model.load_state_dict(torch.load("pytorch_model.bin", map_location="cpu"))
+model.eval()
+
+# Inference function
+def predict_digit(image):
+    image = image.convert("L").resize((28, 28))        # Convert to grayscale
+    image = np.array(image) / 255.0                     # Normalize
+    image = torch.tensor(image).unsqueeze(0).unsqueeze(0).float()  # (1, 1, 28, 28)
+    with torch.no_grad():
+        logits = model(image)
+        probs = F.softmax(logits, dim=1).numpy().flatten()
+        predicted = np.argmax(probs)
+    return {str(i): float(probs[i]) for i in range(10)}
+
+# Gradio UI
+interface = gr.Interface(
+    fn=predict_digit,
+    inputs=gr.Image(type="pil", shape=(280, 280), tool="editor"),
+    outputs=gr.Label(num_top_classes=3),
+    title="Handwritten Digit Classifier",
+    description="Draw a digit or upload a digit image."
+)
+
+if __name__ == "__main__":
+    interface.launch()

model.py

@@ -0,0 +1,20 @@
+class CNN(nn.Module):
+    def __init__(self):
+        super(CNN, self).__init__()
+        self.conv1 = nn.Conv2d(1, 32, kernel_size=3)
+        self.pool1 = nn.MaxPool2d(2)     
+        self.conv2 = nn.Conv2d(32, 64, kernel_size=3) 
+        self.pool2 = nn.MaxPool2d(2)  
+        self.flatten = nn.Flatten()
+        self.fc1 = nn.Linear(64 * 5 * 5, 64)
+        self.fc2 = nn.Linear(64, 10)
+
+    def forward(self, x):
+        x = F.relu(self.conv1(x))
+        x = self.pool1(x)
+        x = F.relu(self.conv2(x))
+        x = self.pool2(x)
+        x = self.flatten(x)
+        x = F.relu(self.fc1(x))
+        x = self.fc2(x)
+        return x 

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7606aba6abcf3685003ebbc6d8dcd6a40189edd71a393a4f46df56d00fb6146e
+size 491361

requirements.txt

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