JLLL0724/garbage_classification_resnet34

1from fastai.vision.all import *2from io import BytesIO3import requests4import streamlit as st5 6"""7# HeartNet8This is a classifier for images of 12-lead EKGs.  It will attempt to detect whether the EKG indicates an acute MI.  It was trained on simulated images.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    ## This **{'is ' if pred == 'mi' else 'is not'}** an MI (heart attack).18    ### Rediction result: {pred}19    ### Probability of {pred}: {probs[key].item()*100: .2f}%20    """21 22 23path = "./"24learn_inf = load_learner(path + "resnet34_stage-3.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