Update app.py

app.py

@@ -21,23 +21,11 @@ import spaces
 import pytesseract
 from PIL import Image
 import torch
-
-zero = torch.Tensor([0]).cuda()
-
-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
+# @spaces.GPU
 def ocr_with_paddle(img):
     print(zero.device) 
     finaltext = ''
@@ -54,7 +42,7 @@ def ocr_with_paddle(img):
 """
 Keras OCR
 """
-@spaces.GPU
+# @spaces.GPU
 def ocr_with_keras(img):
     print(zero.device) 
     output_text = ''
@@ -77,7 +65,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)
@@ -90,6 +78,7 @@ def ocr_with_easy(img):
 """
 Generate OCR
 """
+@spaces.GPU
 def generate_ocr(Method,img):
     
     text_output = ''