Update app.py

app.py

@@ -18,7 +18,15 @@ from PIL import Image
 from paddleocr import PaddleOCR
 from save_data import flag
 import spaces
-            
+
+gpus = tf.config.experimental.list_physical_devices('GPU')
+if gpus:
+    try:
+        for gpu in gpus:
+            tf.config.experimental.set_memory_growth(gpu, True)
+    except RuntimeError as e:
+        print(e)
+        
 """
 Paddle OCR
 """
@@ -59,6 +67,8 @@ def get_grayscale(image):
 # Thresholding or Binarization
 def thresholding(src):
     return cv2.threshold(src,127,255, cv2.THRESH_TOZERO)[1]
+
+@spaces.GPU
 def ocr_with_easy(img):
     gray_scale_image=get_grayscale(img)
     thresholding(gray_scale_image)