# AUTOGENERATED! DO NOT EDIT! File to edit: minst.ipynb.

# %% auto 0
__all__ = ['base_path', 'image_path']

# %% minst.ipynb 63
from fastai.vision.all import *
import streamlit as st

# %% minst.ipynb 64
base_path = './mnist_png/'
image_path = base_path + 'testing/0/3.png'
st.image(image_path)

# %% minst.ipynb 65
if st.button('Predict'):
    learn_inf = load_learner(base_path + 'export.pkl')
    res = learn_inf.predict(image_path)
    st.write(f"Output: {res[0]}")