x10101/example

1from fastai.vision.all import *2from io import BytesIO3import requests4import streamlit as st5 6"""7# 測試用8"""9 10def predict(img):11    st.image(img, caption="Your image", use_column_width=True)12    pred, key, probs = learn_inf.predict(img)13    # st.write(learn_inf.predict(img))14 15    f"""16    ### Rediction result: {pred}17    ### Probability of {pred}: {probs[key].item()*100: .2f}%18    """19 20 21path = "./"22learn_inf = load_learner(path + "demo_model.pkl")23 24option = st.radio("", ["Upload Image", "Image URL"])25 26if option == "Upload Image":27    uploaded_file = st.file_uploader("Please upload an image.")28 29    if uploaded_file is not None:30        img = PILImage.create(uploaded_file)31        predict(img)32 33else:34    url = st.text_input("Please input a url.")35 36    if url != "":37        try:38            response = requests.get(url)39            pil_img = PILImage.create(BytesIO(response.content))40            predict(pil_img)41 42        except:43            st.text("Problem reading image from", url)44