Update app.py

app.py

@@ -21,22 +21,26 @@ import spaces
 import pytesseract
 from PIL import Image
 
-os.environ['CUDA_VISIBLE_DEVICES'] = "1"
+zero = torch.Tensor([0]).cuda()
+tf.config.experimental.set_memory_growth(zero, True)
 
-gpus = tf.config.experimental.list_physical_devices('GPU')
-print("\n\n gpus  :",gpus)
-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'] = zero
+print(zero.device) 
+# gpus = tf.config.experimental.list_physical_devices('GPU')
+# print("\n\n gpus  :",gpus)
+# if gpus:
+#     try:
+#         for gpu in gpus:
+#             tf.config.experimental.set_memory_growth(gpu, True)
+#     except RuntimeError as e:
+#         print(e)
         
 """
 Paddle OCR
 """
 @spaces.GPU
 def ocr_with_paddle(img):
+    print(zero.device) 
     finaltext = ''
     ocr = PaddleOCR(use_gpu=True,lang='en',use_angle_cls=True)
     # img_path = 'exp.jpeg'
@@ -53,6 +57,7 @@ Keras OCR
 """
 @spaces.GPU
 def ocr_with_keras(img):
+    print(zero.device) 
     output_text = ''
     pipeline=keras_ocr.pipeline.Pipeline()
     images=[keras_ocr.tools.read(img)]