Spaces:
Sleeping
Sleeping
Update src/OCR.py
Browse files- src/OCR.py +2 -2
src/OCR.py
CHANGED
|
@@ -8,7 +8,6 @@ from utils.qa import chain
|
|
| 8 |
from langchain.memory import ConversationBufferWindowMemory
|
| 9 |
from langchain_community.chat_message_histories import StreamlitChatMessageHistory
|
| 10 |
|
| 11 |
-
reader = easyocr.Reader(['en', 'hi'], gpu=False)
|
| 12 |
def I_OCR():
|
| 13 |
# Function to display the OCR image with bounding boxes and text
|
| 14 |
def display_ocr_image(img, results):
|
|
@@ -46,7 +45,8 @@ def I_OCR():
|
|
| 46 |
if file is not None:
|
| 47 |
image = Image.open(file)
|
| 48 |
st.image(image)
|
| 49 |
-
|
|
|
|
| 50 |
results = reader.readtext(np.array(image))
|
| 51 |
|
| 52 |
img_df = pd.DataFrame(results, columns=['bbox', 'Predicted Text', 'Prediction Confidence'])
|
|
|
|
| 8 |
from langchain.memory import ConversationBufferWindowMemory
|
| 9 |
from langchain_community.chat_message_histories import StreamlitChatMessageHistory
|
| 10 |
|
|
|
|
| 11 |
def I_OCR():
|
| 12 |
# Function to display the OCR image with bounding boxes and text
|
| 13 |
def display_ocr_image(img, results):
|
|
|
|
| 45 |
if file is not None:
|
| 46 |
image = Image.open(file)
|
| 47 |
st.image(image)
|
| 48 |
+
|
| 49 |
+
reader = easyocr.Reader(['en', 'hi'], gpu=False)
|
| 50 |
results = reader.readtext(np.array(image))
|
| 51 |
|
| 52 |
img_df = pd.DataFrame(results, columns=['bbox', 'Predicted Text', 'Prediction Confidence'])
|