37 lines
778 B
Python
37 lines
778 B
Python
import argparse
|
|
import json
|
|
|
|
from image_prediction.config import CONFIG
|
|
from image_prediction.pipeline import load_pipeline
|
|
from image_prediction.utils import get_logger
|
|
import logging
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("pdf")
|
|
|
|
args = parser.parse_args()
|
|
|
|
return args
|
|
|
|
|
|
def main(args):
|
|
pipeline = load_pipeline(verbose=True)
|
|
|
|
with open(args.pdf, "rb") as f:
|
|
predictions = pipeline(f.read())
|
|
|
|
for prd in predictions:
|
|
print(json.dumps(prd, indent=2))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# logging.basicConfig(level=logging.DEBUG)
|
|
logger = get_logger()
|
|
|
|
logger.info("111111111111111111111111111111111")
|
|
print(logger.level)
|
|
print(logging.DEBUG)
|
|
args = parse_args()
|
|
# main(args)
|