35 lines
1009 B
Python
35 lines
1009 B
Python
import argparse
|
|
import json
|
|
import requests
|
|
|
|
from vidocp.utils.preprocessing import open_pdf
|
|
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--pdf_path", required=True, help="path to PDF file")
|
|
parser.add_argument("--first_page", type=int, required=True, help="path to PDF file")
|
|
parser.add_argument("--last_page", type=int, required=False, default=None, help="path to PDF file")
|
|
args = parser.parse_args()
|
|
|
|
return args
|
|
|
|
|
|
def main(args):
|
|
|
|
#data = open_pdf(args.pdf_path, args.first_page, args.last_page)
|
|
# params = json.dumps({
|
|
# "pdf_path": "a",#args.pdf_path,
|
|
# "first_page": 4,#args.first_page,
|
|
# "last_page": 6#args.last_page
|
|
# })
|
|
response = requests.post("http://127.0.0.1:5000", data=open(args.pdf_path, "rb"))#, json=params)
|
|
response.raise_for_status()
|
|
predictions = response.json()
|
|
|
|
print(json.dumps(predictions, indent=2))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
args = parse_args()
|
|
main(args) |