update spaces

app.py

@@ -0,0 +1,58 @@
+import time
+
+import gradio as gr
+import torch
+import torch.nn.functional as F
+from huggingface_hub import hf_hub_download
+
+from digitnet import Model
+
+torch.set_grad_enabled(False)
+
+hf_hub_download("karanravindra/digitnet", filename="mnist", local_dir=".")
+
+model = Model(1, 16, num_classes=36)
+model.load_state_dict(
+    torch.load("models/emnist-model.pth", weights_only=True, map_location="cpu")
+)
+model.eval()
+
+all_classes = list("0123456789abcdefghijklmnopqrstuvwxyz")
+all_classes = map(str.upper, all_classes)
+
+
+def predict(inputs: dict) -> dict[str, float]:
+    print(f"[{time.strftime('%H:%M:%S')}] Predicting...")
+
+    img = torch.from_numpy(inputs["composite"]).unsqueeze(0).unsqueeze(0).float()
+    img = img / img.max()
+
+    img = F.interpolate(img, size=(32, 32), mode="bilinear", align_corners=False)
+
+    logits = model(img)
+
+    probs = torch.softmax(logits, dim=1).squeeze()
+
+    return {c: p for c, p in zip(all_classes, probs.tolist())}
+
+
+input = gr.Sketchpad(
+    image_mode="L",
+    canvas_size=(600, 600),
+    layers=False,
+    brush=gr.Brush(colors=["rgb(239, 68, 68)"], color_mode="fixed", default_size=20),
+)
+
+output = gr.Label(num_top_classes=10)
+
+demo = gr.Interface(
+    fn=predict,
+    live=True,
+    inputs=input,
+    outputs="label",
+    title="DigitNet",
+    description="A simple handwritten number and letter classifier.\n\nDraw a digit or letter in the box below and see the model's predictions.",
+)
+
+if __name__ == "__main__":
+    demo.launch()

reqirements.txt

@@ -0,0 +1,99 @@
+digitnet @ git+https://github.com/karanravindra/digitnet@main
+
+# This file was autogenerated via `uv export`.
+-e .
+aiofiles==23.2.1
+aiohappyeyeballs==2.4.3
+aiohttp==3.10.10
+aiosignal==1.3.1
+annotated-types==0.7.0
+anyio==4.6.2.post1
+async-timeout==4.0.3 ; python_full_version < '3.11'
+attrs==24.2.0
+certifi==2024.8.30
+charset-normalizer==3.4.0
+click==8.1.7
+colorama==0.4.6 ; platform_system == 'Windows'
+docker-pycreds==0.4.0
+exceptiongroup==1.2.2 ; python_full_version < '3.11'
+fastapi==0.115.2
+ffmpy==0.4.0
+filelock==3.16.1
+frozenlist==1.4.1
+fsspec==2024.10.0
+gitdb==4.0.11
+gitpython==3.1.43
+gradio==5.1.0
+gradio-client==1.4.0
+h11==0.14.0
+httpcore==1.0.6
+httpx==0.27.2
+huggingface-hub==0.26.0
+idna==3.10
+jinja2==3.1.4
+lightning==2.4.0
+lightning-utilities==0.11.8
+markdown-it-py==3.0.0 ; sys_platform != 'emscripten'
+markupsafe==2.1.5
+mdurl==0.1.2 ; sys_platform != 'emscripten'
+mpmath==1.3.0
+multidict==6.1.0
+networkx==3.4.1
+numpy==1.26.4
+nvidia-cublas-cu12==12.4.5.8 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+nvidia-cuda-cupti-cu12==12.4.127 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+nvidia-cuda-nvrtc-cu12==12.4.127 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+nvidia-cuda-runtime-cu12==12.4.127 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+nvidia-cudnn-cu12==9.1.0.70 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+nvidia-cufft-cu12==11.2.1.3 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+nvidia-curand-cu12==10.3.5.147 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+nvidia-cusolver-cu12==11.6.1.9 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+nvidia-cusparse-cu12==12.3.1.170 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+nvidia-nccl-cu12==2.21.5 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+nvidia-nvjitlink-cu12==12.4.127 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+nvidia-nvtx-cu12==12.4.127 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+orjson==3.10.9
+packaging==24.1
+pandas==2.2.3
+pillow==10.4.0
+platformdirs==4.3.6
+propcache==0.2.0
+protobuf==5.28.2
+psutil==6.1.0
+pydantic==2.9.2
+pydantic-core==2.23.4
+pydub==0.25.1
+pygments==2.18.0 ; sys_platform != 'emscripten'
+python-dateutil==2.9.0.post0
+python-multipart==0.0.12
+pytorch-lightning==2.4.0
+pytz==2024.2
+pyyaml==6.0.2
+requests==2.32.3
+rich==13.9.2 ; sys_platform != 'emscripten'
+ruff==0.7.0 ; sys_platform != 'emscripten'
+semantic-version==2.10.0
+sentry-sdk==2.17.0
+setproctitle==1.3.3
+setuptools==75.2.0
+shellingham==1.5.4 ; sys_platform != 'emscripten'
+six==1.16.0
+smmap==5.0.1
+sniffio==1.3.1
+starlette==0.40.0
+sympy==1.13.1
+tomlkit==0.12.0
+torch==2.5.0
+torchinfo==1.8.0
+torchmetrics==1.5.0
+torchvision==0.20.0
+tqdm==4.66.5
+triton==3.1.0 ; python_full_version < '3.13' and platform_machine == 'x86_64' and platform_system == 'Linux'
+typer==0.12.5 ; sys_platform != 'emscripten'
+typing-extensions==4.12.2
+tzdata==2024.2
+urllib3==2.2.3
+uvicorn==0.32.0 ; sys_platform != 'emscripten'
+wandb==0.18.5
+websockets==12.0
+yarl==1.15.5