Update app classifacation.py

app classifacation.py

@@ -5,7 +5,7 @@ import streamlit as st
 
 """
 # HeartNet
-This 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.
+This is a classifier for images of tzuyu iu hanni and winter.  
 """
 
 def predict(img):
@@ -14,14 +14,13 @@ def predict(img):
     # st.write(learn_inf.predict(img))
 
     f"""
-    ## This **{'is ' if pred == 'mi' else 'is not'}** an MI (heart attack).
     ### Rediction result: {pred}
     ### Probability of {pred}: {probs[key].item()*100: .2f}%
     """
 
 
 path = "./"
-learn_inf = load_learner(path + "demo_model.pkl")
+learn_inf = load_learner(path + "resnet34_stage-1cele.pkl")
 
 option = st.radio("", ["Upload Image", "Image URL"])