Update app.py

app.py

@@ -8,6 +8,7 @@ import gradio as gr
 from paddleocr import PaddleOCR
 import psutil
 import time
+import spaces
 
 ocr = PaddleOCR(use_angle_cls=True, lang='en', det_model_dir='models/det', rec_model_dir='models/rec', cls_model_dir='models/cls')
 
@@ -41,6 +42,7 @@ def sample_border_color(image, box, padding=2):
     median_color = np.median(border_pixels, axis=0)
     return tuple(map(int, median_color))
 
+@spaces.GPU    
 def detect_text_boxes(image):
     results = ocr.ocr(image, cls=True)
     boxes = []