-
Notifications
You must be signed in to change notification settings - Fork 6
/
parser.py
26 lines (22 loc) · 782 Bytes
/
parser.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
from predictLM import predict
import yaml
from objectify import Struct
class Parser:
def __init__(self, pdf_path, pdf_name):
self.config_path = "config.yaml"
self.pdf_path = pdf_path
self.pdf_name = pdf_name
def checkbox_parser(self):
with open(self.config_path) as file:
model_config = yaml.load(file)
model_info_dict = {}
for key in model_config.keys():
model_info_dict.update(model_config[key])
args = Struct(**model_info_dict)
args.pdf_path = self.pdf_path
args.pdf_name = self.pdf_name
df_processed = predict(args)
return df_processed
img_path = "/home/nitheesh/Downloads/image/front/EHE2.png"
obj = Parser(img_path,"")
obj.checkbox_parser()