EmilyCh/Seals

1from fastai.vision.all import *2from io import BytesIO3import requests4import streamlit as st5 6"""7# 分辨海獅的兄弟姊妹8eyyyy~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    ### 該生物與{pred}有{probs[key].item()*100: .2f}%相似18    """19 20path = "./"21learn_inf = load_learner(path + "resnet34_stage-2.pkl")22 23option = st.radio("", ["Upload Image", "Image URL"])24 25if option == "Upload Image":26    uploaded_file = st.file_uploader("Please upload an image.")27 28    if uploaded_file is not None:29        img = PILImage.create(uploaded_file)30        predict(img)31 32else:33    url = st.text_input("Please input a url.")34 35    if url != "":36        try:37            response = requests.get(url)38            pil_img = PILImage.create(BytesIO(response.content))39            predict(pil_img)40 41        except:42            st.text("Problem reading image from", url)43