Update app.py

app.py

@@ -1,27 +1,29 @@
+import tempfile
+import os
 import streamlit as st
-import paddleocr
+import paddlehub as hub
+from PIL import Image
+import io
 
-# Set up the OCR model
-model = paddleocr.create_model(mode='rec', lang='ch_sim')
+pp_ocrv3 = hub.Module(name="ch_pp-ocrv3")
 
-def ocr_image(image):
-  # Perform OCR on the image
-  text = model.recognize(image)
+def inference(img):
+  with tempfile.TemporaryDirectory() as tempdir_name:
+    pp_ocrv3.recognize_text(paths=[img],use_gpu=False,output_dir=tempdir_name,visualization=True)
+    result_names = os.listdir(tempdir_name)
+    output_image = Image.open(os.path.join(tempdir_name, result_names[0]))
+    return [output_image]
+  
+title="ch_PP-OCRv3"
+description="ch_PP-OCRv3 is a practical ultra-lightweight OCR system developed by PaddleOCR."
 
-  return text
+examples=[['test.png']]
 
-# Set up the Streamlit app
-st.title('OCR App')
-
-# Add a file upload widget
-uploaded_file = st.file_uploader('Choose an image file')
 
+uploaded_file = st.file_uploader("Choose a file")
 if uploaded_file is not None:
-  # Read the image file
-  image = Image.open(uploaded_file)
-
-  # Perform OCR on the image
-  text = ocr_image(image)
-
-  # Display the OCR results
-  st.text(text)
+    # To read file as bytes:
+    bytes_data = uploaded_file.getvalue()
+    st.write(bytes_data)
+    image = Image.open(io.BytesIO(bytes_data))
+    st.image(image, caption='Sunrise by the mountains')