Update app.py

app.py

@@ -21,13 +21,15 @@ import spaces
 import pytesseract
 from PIL import Image
 
-# 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)
+os.environ['CUDA_VISIBLE_DEVICES'] = "0"
+
+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
@@ -48,6 +50,7 @@ def ocr_with_paddle(img):
 """
 Keras OCR
 """
+@spaces.GPU
 def ocr_with_keras(img):
     output_text = ''
     pipeline=keras_ocr.pipeline.Pipeline()