Update app.py

app.py

@@ -11,6 +11,11 @@ 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"""
+    
+    ### Rediction result: {pred}
+    ### Probability of {pred}: {probs[key].item()*100: .2f}%
+    """