first commit

.gitignore

@@ -0,0 +1 @@
+env/

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "python.formatting.provider": "black"
+}

app.py

@@ -0,0 +1,55 @@
+import pickle
+
+import cv2
+import joblib
+import numpy as np
+import streamlit as st
+import torch
+from streamlit_drawable_canvas import st_canvas
+
+BINARY = joblib.load("model.joblib")
+ML_MODEL = pickle.loads(BINARY)
+ML_MODEL.load_state_dict(
+    torch.load("model_weights.pth", map_location=torch.device("cpu"))
+)
+ML_MODEL.eval()
+
+
+def predict_number(img):
+    if img is None:
+        return None, None
+
+    inp = torch.tensor(img, dtype=torch.float32).unsqueeze(0).unsqueeze(0) / 255.0
+
+    with torch.no_grad():
+        output = ML_MODEL(inp)
+
+    probabilities = torch.nn.functional.softmax(output[0], dim=0)
+    values, indices = torch.topk(probabilities, 5)
+    confidences = {f"is number {i.item()}": v.item() for i, v in zip(indices, values)}
+
+    return confidences
+
+
+canvas_result = st_canvas(
+    fill_color="rgba(255, 165, 0, 0.3)",
+    stroke_width=17,
+    stroke_color="#000000",
+    background_color="#ffffff",
+    background_image=None,
+    update_streamlit=True,
+    height=200,
+    width=200,
+    drawing_mode="freedraw",
+    key="canvas",
+)
+
+if canvas_result.image_data is not None:
+    image_data = canvas_result.image_data[:, :, 0]
+    image_data = np.squeeze(image_data)
+    image_data = cv2.blur(image_data, (10, 10))
+    image_data = cv2.resize(image_data, (28, 28))
+    image_data = cv2.bitwise_not(image_data)
+    st.image(image_data)
+    confidences = predict_number(image_data)
+    st.write(confidences)

model.joblib

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

model_weights.pth

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

requirements.in

@@ -0,0 +1,7 @@
+torch
+torchvision
+joblib
+streamlit
+streamlit-drawable-canvas
+opencv-python
+cloudpickle

requirements.txt

@@ -0,0 +1,136 @@
+#
+# This file is autogenerated by pip-compile with Python 3.10
+# by the following command:
+#
+#    pip-compile
+#
+altair==4.2.2
+    # via streamlit
+attrs==22.2.0
+    # via jsonschema
+blinker==1.5
+    # via streamlit
+cachetools==5.3.0
+    # via streamlit
+certifi==2022.12.7
+    # via requests
+charset-normalizer==3.0.1
+    # via requests
+click==8.1.3
+    # via streamlit
+cloudpickle==2.2.1
+    # via -r requirements.in
+decorator==5.1.1
+    # via validators
+entrypoints==0.4
+    # via altair
+gitdb==4.0.10
+    # via gitpython
+gitpython==3.1.31
+    # via streamlit
+idna==3.4
+    # via requests
+importlib-metadata==6.0.0
+    # via streamlit
+jinja2==3.1.2
+    # via
+    #   altair
+    #   pydeck
+joblib==1.2.0
+    # via -r requirements.in
+jsonschema==4.17.3
+    # via altair
+markdown-it-py==2.2.0
+    # via rich
+markupsafe==2.1.2
+    # via jinja2
+mdurl==0.1.2
+    # via markdown-it-py
+numpy==1.24.2
+    # via
+    #   altair
+    #   opencv-python
+    #   pandas
+    #   pyarrow
+    #   pydeck
+    #   streamlit
+    #   streamlit-drawable-canvas
+    #   torchvision
+opencv-python==4.7.0.72
+    # via -r requirements.in
+packaging==23.0
+    # via streamlit
+pandas==1.5.3
+    # via
+    #   altair
+    #   streamlit
+pillow==9.4.0
+    # via
+    #   streamlit
+    #   streamlit-drawable-canvas
+    #   torchvision
+protobuf==3.20.3
+    # via streamlit
+pyarrow==11.0.0
+    # via streamlit
+pydeck==0.8.0
+    # via streamlit
+pygments==2.14.0
+    # via rich
+pympler==1.0.1
+    # via streamlit
+pyrsistent==0.19.3
+    # via jsonschema
+python-dateutil==2.8.2
+    # via
+    #   pandas
+    #   streamlit
+pytz==2022.7.1
+    # via pandas
+pytz-deprecation-shim==0.1.0.post0
+    # via tzlocal
+requests==2.28.2
+    # via
+    #   streamlit
+    #   torchvision
+rich==13.3.1
+    # via streamlit
+semver==2.13.0
+    # via streamlit
+six==1.16.0
+    # via python-dateutil
+smmap==5.0.0
+    # via gitdb
+streamlit==1.19.0
+    # via
+    #   -r requirements.in
+    #   streamlit-drawable-canvas
+streamlit-drawable-canvas==0.9.2
+    # via -r requirements.in
+toml==0.10.2
+    # via streamlit
+toolz==0.12.0
+    # via altair
+torch==1.13.1
+    # via
+    #   -r requirements.in
+    #   torchvision
+torchvision==0.14.1
+    # via -r requirements.in
+tornado==6.2
+    # via streamlit
+typing-extensions==4.5.0
+    # via
+    #   streamlit
+    #   torch
+    #   torchvision
+tzdata==2022.7
+    # via pytz-deprecation-shim
+tzlocal==4.2
+    # via streamlit
+urllib3==1.26.14
+    # via requests
+validators==0.20.0
+    # via streamlit
+zipp==3.15.0
+    # via importlib-metadata