esther11021/Tree1

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