Update app.py

app.py

@@ -17,13 +17,16 @@ from datasets import load_dataset, Image
 from PIL import Image
 from paddleocr import PaddleOCR
 from save_data import flag
+import spaces
             
 """
 Paddle OCR
 """
+@spaces.GPU
 def ocr_with_paddle(img):
     finaltext = ''
-    ocr = PaddleOCR(lang='en', use_angle_cls=True)
+    paddle.set_device('gpu') 
+    ocr = PaddleOCR(lang='en',use_angle_cls=True)
     # img_path = 'exp.jpeg'
     result = ocr.ocr(img)
     
@@ -32,6 +35,7 @@ def ocr_with_paddle(img):
         finaltext += ' '+ text
     return finaltext
 
+    
 """
 Keras OCR
 """