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 |
|