Update app.py

app.py

@@ -4,24 +4,18 @@ import requests
 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.
+# Sustainable energy classifacation
+This is a classifier for images ofSustainable energy.   
 """
-
 def predict(img):
     st.image(img, caption="Your image", use_column_width=True)
     pred, key, probs = learn_inf.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-1.pkl")
 
 option = st.radio("", ["Upload Image", "Image URL"])