Spaces:
Paused
Paused
import argparse | |
from asyncio.log import logger | |
from Layoutlm_inference.ocr import prepare_batch_for_inference | |
from Layoutlm_inference.inference_handler import handle | |
import logging | |
import os | |
if __name__ == "__main__": | |
# try: | |
parser = argparse.ArgumentParser() | |
parser.add_argument("--model_path", type=str) | |
parser.add_argument("--images_path", type=str) | |
args, _ = parser.parse_known_args() | |
images_path = args.images_path | |
image_files = os.listdir(images_path) | |
images_path = [images_path+f'/{image_file}' for image_file in image_files] | |
inference_batch = prepare_batch_for_inference(images_path) | |
context = {"model_dir": args.model_path} | |
output_ls = handle(inference_batch,context) | |
print("output_ls",output_ls) | |
# except Exception as err: | |
# os.makedirs('log', exist_ok=True) | |
# logging.basicConfig(filename='log/error_output.log', level=logging.ERROR, | |
# format='%(asctime)s %(levelname)s %(name)s %(message)s') | |
# logger = logging.getLogger(__name__) | |
# logger.error(err) | |