Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ import gradio as gr
|
|
8 |
from paddleocr import PaddleOCR
|
9 |
import psutil
|
10 |
import time
|
11 |
-
import spaces
|
12 |
|
13 |
ocr = PaddleOCR(use_angle_cls=True, lang='en', det_model_dir='models/det', rec_model_dir='models/rec', cls_model_dir='models/cls')
|
14 |
|
@@ -104,7 +104,7 @@ def remove_text_dynamic_fill(img_path, output_path):
|
|
104 |
image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
|
105 |
cv2.imwrite(output_path, image)
|
106 |
|
107 |
-
|
108 |
def process_folder(input_files):
|
109 |
temp_output = tempfile.mkdtemp()
|
110 |
|
|
|
8 |
from paddleocr import PaddleOCR
|
9 |
import psutil
|
10 |
import time
|
11 |
+
#import spaces
|
12 |
|
13 |
ocr = PaddleOCR(use_angle_cls=True, lang='en', det_model_dir='models/det', rec_model_dir='models/rec', cls_model_dir='models/cls')
|
14 |
|
|
|
104 |
image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
|
105 |
cv2.imwrite(output_path, image)
|
106 |
|
107 |
+
#@spaces.GPU
|
108 |
def process_folder(input_files):
|
109 |
temp_output = tempfile.mkdtemp()
|
110 |
|