Update Image Augmentation.py

Image Augmentation.py

@@ -101,7 +101,9 @@ if uploaded_file:
         st.markdown("### Download Transformed Image:")
         download_format = st.radio("Select Download Format:", ["JPG", "PNG", "PDF"])
 
-        if download_format == "JPG" or download_format == "PNG":
+        if download_format == "JPG":
+            st.markdown(download_image(transformed_image, download_format), unsafe_allow_html=True)
+        elif download_format == "PNG":
             st.markdown(download_image(transformed_image, download_format), unsafe_allow_html=True)
         elif download_format == "PDF":
             st.markdown(download_pdf(transformed_image), unsafe_allow_html=True)