Spaces:
Runtime error
Runtime error
| import streamlit as st | |
| from fastai.vision.all import * | |
| from PIL import Image | |
| import pathlib | |
| import urllib.request | |
| temp = pathlib.PosixPath | |
| plt = platform.system() | |
| if plt == 'Windows': pathlib.PosixPath = pathlib.WindowsPath | |
| # MODEL_URL = "https://drive.google.com/uc?export=download&id=1cH5nY1T5oykEcLyWtjA8Wv-Xn8vO20BS" | |
| # urllib.request.urlretrieve(MODEL_URL, "model.pkl") | |
| path = Path() | |
| path.ls(file_exts='.pkl') | |
| learn_inf = load_learner(path/'export.pkl', cpu=True) | |
| # out_pl = st.image(load_image(image), width=250) | |
| def load_image(img_file): | |
| img = PILImage.create(img_file) | |
| return img | |
| def on_click_classify(image): | |
| # load_image(image) | |
| out_pl = st.image(load_image(image), width=250) | |
| pred, pred_idx, probs = learn_inf.predict(load_image(image)) | |
| st.write('Prediction: ', str(pred)[10:], '; Probability: ', float(probs[pred_idx])) | |
| st.title('Dog Classifier') | |
| st.header('Choose your Dog!!') | |
| image = st.file_uploader(label=' ', type=['png', 'jpg'], key='img', help='upload an img of dog') | |
| # picture = st.camera_input(label='Click your dog!') | |
| # btn_run.on_change(on_click_classify) | |
| btn_run = st.button(label='Classify') | |
| if btn_run: | |
| on_click_classify(image) | |
| st.markdown('#### Created by **Umang Kaushik**') | |
| st.markdown('##### **[Github](https://github.com/Umang-10)**') | |