Spaces:
Runtime error
Runtime error
File size: 1,320 Bytes
8826d5f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
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)**')
|