Update app.py

app.py

@@ -675,7 +675,7 @@ def main():
         if st.session_state.predictions is not None:
             for idx, predictions in st.session_state.predictions:
                 st.markdown(f"### Region {idx}")
-
+                st.image(st.session_state.image[idx], use_column_width=True)
                 # Display main prediction
                 top_class, top_confidence = predictions[0]
                 st.markdown(f"**Primary Prediction: Grade {top_class}**")