kermitt2 / grobid

A machine learning software for extracting information from scholarly documents
https://grobid.readthedocs.io
Apache License 2.0
3.27k stars 439 forks source link

How to load deep learning models using tensorflow or keras? #1025

Open himanshu-aptara opened 1 year ago

himanshu-aptara commented 1 year ago

Hi, I am trying to use tensorflow and keras, model_from_json, load model function to load the deep learning models inside citation model, but with no luck. Can you guide me how to load these models in our local notebook?

kermitt2 commented 1 year ago

Hi @himanshu-aptara !

To run the Deep Learning models outside Grobid/Java, you need to use https://github.com/kermitt2/delft

For exemple, running the bibliographical reference parsing model on a few examples:

python3 delft/applications/grobidTagger.py citation tag --architecture BidLSTM_CRF

DeLFT is built on top TensorFlow 2, Keras and transformers libraries.