| # 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]}") | |