pragnakalp commited on
Commit
6d9f1bd
1 Parent(s): a6fc706

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -7
app.py CHANGED
@@ -18,6 +18,8 @@ from PIL import Image
18
  from paddleocr import PaddleOCR
19
  from save_data import flag
20
  import spaces
 
 
21
 
22
  gpus = tf.config.experimental.list_physical_devices('GPU')
23
  if gpus:
@@ -48,13 +50,27 @@ Keras OCR
48
  """
49
  @spaces.GPU
50
  def ocr_with_keras(img):
51
- output_text = ''
52
- pipeline=keras_ocr.pipeline.Pipeline()
53
- images=[keras_ocr.tools.read(img)]
54
- predictions=pipeline.recognize(images)
55
- first=predictions[0]
56
- for text,box in first:
57
- output_text += ' '+ text
 
 
 
 
 
 
 
 
 
 
 
 
 
 
58
  return output_text
59
 
60
  """
 
18
  from paddleocr import PaddleOCR
19
  from save_data import flag
20
  import spaces
21
+ import pytesseract
22
+ from PIL import Image
23
 
24
  gpus = tf.config.experimental.list_physical_devices('GPU')
25
  if gpus:
 
50
  """
51
  @spaces.GPU
52
  def ocr_with_keras(img):
53
+ # output_text = ''
54
+ # pipeline=keras_ocr.pipeline.Pipeline()
55
+ # images=[keras_ocr.tools.read(img)]
56
+ # predictions=pipeline.recognize(images)
57
+ # first=predictions[0]
58
+ # for text,box in first:
59
+ # output_text += ' '+ text
60
+ # return output_text
61
+
62
+
63
+ # Path to Tesseract executable (you may need to change this)
64
+ pytesseract.pytesseract.tesseract_cmd = r'<path_to_tesseract_executable>'
65
+
66
+ # Load the image
67
+ image = Image.open(img)
68
+
69
+ # Perform OCR on the image
70
+ output_text = pytesseract.image_to_string(image)
71
+
72
+ # Print the extracted text
73
+ print(output_text)
74
  return output_text
75
 
76
  """