Update app.py

app.py

@@ -14,11 +14,20 @@ 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}
+    ## This **{'is cordana' if pred == 'cordana' else 'is pestalotiopsis' if pred == 'pestalotiopsis' else 'is sigatoka' if pred == 'sigatoka' else 'is healthy'}**.
+    ### Prediction result: {pred}
     ### Probability of {pred}: {probs[key].item()*100: .2f}%
     """
 
+col1, col2, col3 = st.columns(3)
+
+with col1:
+    st.image(img1, caption="Image 1", use_column_width=True)
+with col2:
+    st.image(img2, caption="Image 2", use_column_width=True)
+with col3:
+    st.image(img3, caption="Image 3", use_column_width=True)
+
 
 path = "./"
 learn_inf = load_learner(path + "demo_model.pkl")