Aqdas commited on
Commit
52bc096
·
verified ·
1 Parent(s): 021202f

Update image_to_text.py

Browse files
Files changed (1) hide show
  1. image_to_text.py +6 -4
image_to_text.py CHANGED
@@ -1,8 +1,10 @@
1
  from PIL import Image
2
  import pytesseract
 
3
 
4
 
5
- def image_to_text(image_path: str) -> str:
6
- image = Image.open(image_path)
7
- extracted_text = pytesseract.image_to_string(image)
8
- return extracted_text
 
 
1
  from PIL import Image
2
  import pytesseract
3
+ import io
4
 
5
 
6
+ def image_to_text(images):
7
+ img_bytes = io.BytesIO()
8
+ images.save(img_bytes, format='PNG')
9
+ text = pytesseract.image_to_string(Image.open(img_bytes))
10
+ return text