Update app.py

app.py

@@ -27,6 +27,20 @@ def get_ocr_model():
     return CnOcr()
 
 
+def save_image(img):
+    from io import BytesIO
+
+    buf = BytesIO()
+    img.save(buf, format="JPEG")
+    byte_im = buf.getvalue()
+    st.download_button(
+        label="下载图片",
+        data=byte_im,
+        file_name="antiOCR.jpeg",
+        mime="image/jpeg",
+    )
+
+
 def main():
     st.sidebar.header('输出设置')
     char_reverse_ratio = st.sidebar.slider(
@@ -118,6 +132,8 @@ def main():
                 )
             st.subheader('输出图片')
             st.image(out_img)
+            save_image(out_img)
+            
             st.markdown('**对输出图片进行OCR，结果如下（如果依旧出现敏感词，尝试重新生成图片）：**')
             ocr_out = ocr.ocr(out_img)
             new_texts = [out['text'] for out in ocr_out]