Spaces:
Running
on
Zero
Running
on
Zero
Revert "initial commit"
Browse filesThis reverts commit 6c3c1e89b3ab981b79a13afc00dc7e6ddc3f7fe0.
app.py
CHANGED
|
@@ -47,8 +47,6 @@ model.config.pad_token_id = tokenizer.eos_token_id
|
|
| 47 |
|
| 48 |
@spaces.GPU
|
| 49 |
def process_image(image, task, ocr_type=None, ocr_box=None, ocr_color=None, render=False):
|
| 50 |
-
# Load the image using loadimg
|
| 51 |
-
img = load_img(image, output_type="pil", input_type="auto")
|
| 52 |
if task == "Plain Text OCR":
|
| 53 |
res = model.chat(tokenizer, image, ocr_type='ocr')
|
| 54 |
elif task == "Format Text OCR":
|
|
|
|
| 47 |
|
| 48 |
@spaces.GPU
|
| 49 |
def process_image(image, task, ocr_type=None, ocr_box=None, ocr_color=None, render=False):
|
|
|
|
|
|
|
| 50 |
if task == "Plain Text OCR":
|
| 51 |
res = model.chat(tokenizer, image, ocr_type='ocr')
|
| 52 |
elif task == "Format Text OCR":
|