Add image_viewer

app.py

@@ -7,6 +7,7 @@ import pathlib
 
 header = st.container()
 inference = st.container()
+image_viewer = st.container()
 
 with header:
 	st.title("Cuisine Classifier")
@@ -33,4 +34,8 @@ with inference:
         pred, pred_idx, probs = learn_inf.predict(img)
         prob_value = probs[pred_idx].item()
         rounded_prob_percentage = round(prob_value * 100)
-        st.text(f"This is {pred}, isn't it? Believe me, I am {rounded_prob_percentage}% sure!")
+        st.text(f"This is {pred}, isn't it? Believe me, I am {rounded_prob_percentage}% sure!")
+
+with image_viewer:
+    st.header(f"Your food pic")
+    st.image(image=img, caption='your pic will be shown here')