32 lines
570 B
Python
32 lines
570 B
Python
import argparse
|
|
import json
|
|
|
|
from image_prediction.pipeline import load_pipeline
|
|
from image_prediction.utils import get_logger
|
|
|
|
|
|
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__":
|
|
logger = get_logger()
|
|
|
|
args = parse_args()
|
|
main(args)
|