27 lines
490 B
Python
27 lines
490 B
Python
import argparse
|
|
import json
|
|
|
|
import requests
|
|
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--pdf_path", required=True)
|
|
args = parser.parse_args()
|
|
|
|
return args
|
|
|
|
|
|
def main(args):
|
|
|
|
response = requests.post("http://127.0.0.1:5000", data=open(args.pdf_path, "rb"))
|
|
response.raise_for_status()
|
|
predictions = response.json()
|
|
|
|
print(json.dumps(predictions, indent=2))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
args = parse_args()
|
|
main(args)
|