Added download button

app.py

@@ -192,6 +192,12 @@ elif app_mode == 'Check on Image':
   st.subheader('Output Image')
   st.image(out_image, channels = 'BGR', use_column_width = True)
 
+  pil_out_image = Image.fromarray(cv2.cvtColor(out_image, cv2.COLOR_BGR2RGB))
+  out_image_bytes = io.BytesIO()
+  pil_out_image.save(out_image_bytes, format='JPEG')
+
+  st.download_button('Download', data=out_image_bytes.getvalue(), file_name='out_image.jpg')
+
 elif app_mode == 'Check on Video':
 
   st.set_option('deprecation.showfileUploaderEncoding', False)