in src/train.py [0:0]
def train():
html_files_path = '/Users/Vbaungally/Downloads/autofill_html_dataset'
artifact_path = 'artifacts' # saves models to this path
ds = generate_dataset(html_files_path, samples=20000)
print(ds)
model = AutofillModel()
model.load_data(ds)
model.train(artifact_path)