Update app.py

app.py

@@ -1,9 +1,9 @@
 # -*- encoding: utf-8 -*-
 # @Author: SWHL
 # @Contact: liekkaskono@163.com
-from pathlib import Path
-
+import numpy as np
 import streamlit as st
+from PIL import Image
 
 
 if __name__ == '__main__':
@@ -18,15 +18,18 @@ if __name__ == '__main__':
     </p>
     """, unsafe_allow_html=True)
 
-    st.markdown('戳这里查看支持模型列表：[PaddleOCR Models](https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.6/doc/doc_ch/models_list.md)')
-
-    default_url = 'e.g. https://paddleocr.bj.bcebos.com/PP-OCRv3/chinese/ch_PP-OCRv3_rec_infer.tar'
-    url = st.text_input('输入模型地址：', help=default_url)
-
-    default_txt_path = 'e.g. https://raw.githubusercontent.com/PaddlePaddle/PaddleOCR/release/2.6/ppocr/utils/ppocr_keys_v1.txt'
-    txt_url = st.text_input('输入txt地址（文本识别模型时，必选）：', help=default_txt_path)
-
-    save_dir = 'models'
-    save_onnx_path = ''
-
-    is_convert = st.button('Convert')
+    img_suffix = ["png", "jpg", "jpeg"]
+
+    st.markdown('##### rapid-orientation (文档图像分类) demo')
+    img_empty = st.empty()
+    img_file_buffer = st.file_uploader("Upload an image",
+                                       type=img_suffix,
+                                       label_visibility='hidden')
+    if img_file_buffer:
+        image = Image.open(img_file_buffer)
+        img_array = np.array(image)
+        if image:
+            img_empty.image(image, use_column_width=True)
+
+    st.markdown('##### rapid-layout (版面分析) demo')
+    st.markdown('##### rapid-table (表格还原) demo')