Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -42,7 +42,7 @@ def sample_border_color(image, box, padding=2):
|
|
| 42 |
median_color = np.median(border_pixels, axis=0)
|
| 43 |
return tuple(map(int, median_color))
|
| 44 |
|
| 45 |
-
|
| 46 |
def detect_text_boxes(image):
|
| 47 |
results = ocr.ocr(image, cls=True)
|
| 48 |
boxes = []
|
|
@@ -104,6 +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 |
def process_folder(input_files):
|
| 108 |
temp_output = tempfile.mkdtemp()
|
| 109 |
|
|
|
|
| 42 |
median_color = np.median(border_pixels, axis=0)
|
| 43 |
return tuple(map(int, median_color))
|
| 44 |
|
| 45 |
+
|
| 46 |
def detect_text_boxes(image):
|
| 47 |
results = ocr.ocr(image, cls=True)
|
| 48 |
boxes = []
|
|
|
|
| 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 |
|