Update app.py

app.py

@@ -1,23 +1,30 @@
 import streamlit as st
-import paddleocr
+import paddlehub as hub
 
-# Initialize the OCR engine
-engine = paddleocr.create_engine()
+# Load the PaddleOCR module
+module = hub.Module(name='paddleocr')
 
+# Set the OCR detection language to English
+module.set_encode('utf-8')
+module.set_decode('utf-8')
+
+# Create a function to perform OCR on an image
 def ocr(image):
-  # Perform OCR on the image
-  text = engine.run(image)
+  # Use the PaddleOCR module to detect text in the image
+  result = module.detection(images=[image])
+
+  # Extract the text from the OCR result
+  text = result[0]['data'][0]['text']
 
-  # Display the OCR results
-  st.markdown(f'**OCR Output:**\n{text}')
+  return text
 
 # Set up the Streamlit app
-st.title('Image OCR')
+st.title('Image OCR with PaddleOCR')
 
-# Add a file uploader to the app
-image = st.file_uploader('Choose an image')
+# Allow the user to upload an image
+image = st.file_uploader('Choose an image to OCR:')
 
-# If an image is uploaded, display it and perform OCR on it
+# Perform OCR on the image and display the result
 if image is not None:
-  st.image(image, caption='Input Image')
-  ocr(image)
+  text = ocr(image)
+  st.write(f'OCR result: {text}')