Update app.py

app.py

@@ -17,15 +17,15 @@ from datasets import load_dataset, Image
 from PIL import Image
 from paddleocr import PaddleOCR
 from save_data import flag
-# import spaces
+import spaces
 import pytesseract
 from PIL import Image
-# import torch
+import torch
  
 """
 Paddle OCR
 """
-# @spaces.GPU
+@spaces.GPU
 def ocr_with_paddle(img):
     finaltext = ''
     ocr = PaddleOCR(use_gpu=True,lang='en',use_angle_cls=True)
@@ -41,10 +41,10 @@ def ocr_with_paddle(img):
 """
 Keras OCR
 """
-# @spaces.GPU
+@spaces.GPU
 def ocr_with_keras(img):
     output_text = ''
-    pipeline=keras_ocr.pipeline.Pipeline()
+    pipeline=keras_ocr.pipeline.Pipeline().to_device()
     images=[keras_ocr.tools.read(img)]
     predictions=pipeline.recognize(images)
     first=predictions[0]
@@ -63,7 +63,7 @@ def get_grayscale(image):
 def thresholding(src):
     return cv2.threshold(src,127,255, cv2.THRESH_TOZERO)[1]
 
-# @spaces.GPU
+@spaces.GPU
 def ocr_with_easy(img):
     gray_scale_image=get_grayscale(img)
     thresholding(gray_scale_image)
@@ -76,12 +76,9 @@ def ocr_with_easy(img):
 """
 Generate OCR
 """
-# @spaces.GPU
 def generate_ocr(Method,img):
     
     text_output = ''
-    print("\n\n img :",img.any())
-    print("\n\n img.any :",(img).any())
     if img.any() or (img).any():
         add_csv = []
         image_id = 1