Added ability to use custom label map with custom models

allenai / pawls

Software that makes labeling PDFs easy.

https://pawls.apps.allenai.org

Apache License 2.0

380 stars 74 forks source link

Added ability to use custom label map with custom models #131

Closed tjaffri closed 3 years ago

tjaffri commented 3 years ago

Followup to earlier PR #130 this now adds the ability to specify a custom label map for your custom model (if it has a different set of labels than the publaynet detectron2 model).

lolipopshock commented 3 years ago

LGTM!