18 lines
568 B
Python
18 lines
568 B
Python
import json
|
|
|
|
from cv_analysis.server.pipeline import make_image_analysis_pipeline
|
|
from cv_analysis.table_inference import infer_lines
|
|
from cv_analysis.utils.annotate import annotate_pdf
|
|
|
|
pipe = make_image_analysis_pipeline(infer_lines)
|
|
|
|
# FIXME: Implement argparsing
|
|
|
|
pdf_bytes = open("test/test_data/article.pdf", "rb").read()
|
|
vlp_output = json.load(open("test/test_data/article.json", "r"))
|
|
|
|
best_result = list(pipe(data={"pdf": pdf_bytes, "vlp_output": vlp_output}))
|
|
|
|
# print(best_result)
|
|
|
|
annotate_pdf(pdf_bytes, best_result, output_path = "/tmp/deine-mutter.pdf") |