alphagov / govuk-content-metadata

GovNER: an encoder-based language model (RoBERTa) fine-tuned to perform Named Entity Recognition (NER) on GOV.UK content
MIT License
4 stars 1 forks source link

updated fast api model serving in fast_api_model_serving/main.py #100

Closed exfalsoquodlibet closed 1 year ago

exfalsoquodlibet commented 1 year ago

to also include part_of_page as input and output field. This will simplify the post-processing of the Vertex AI batch-prediction pipeline.

Summary

Add your summary here - keep it brief, to the point, and in plain English. For further information about pull requests, check out the GDS Way.

Checklists

This pull/merge request meets the following requirements:

Comments have been added below around the incomplete checks.