Spaces:
Build error
Build error
| from ResumeReader import ResumeReader | |
| from ResumeParser import ResumeParser | |
| from Models import Models | |
| import json | |
| import os | |
| class Main: | |
| def __init__(self): | |
| models = Models() | |
| ner, ner_dates, zero_shot_classifier, tagger = models.load_trained_models() | |
| self.reader = ResumeReader() | |
| self.parser = ResumeParser(ner, ner_dates, zero_shot_classifier, tagger) | |
| def parse_cv(self, file_path): | |
| resume_lines = self.reader.read_file(file_path) | |
| output = self.parser.parse(resume_lines) | |
| return output | |
| def save_parse_as_json(self, dict, file_name): | |
| print("Saving the parse...") | |
| with open(file_name, 'w', encoding="utf-8") as f: | |
| json.dump(dict, f, indent=4, default=str, ensure_ascii=False) |