cyk1337 / bert-text-classification-j Goto Github PK

my Exception:

Starting to predict!
====================== Start loading models ... ======================
====================== Models loaded! ======================
java.lang.NullPointerException
at com.google.common.io.ByteStreams.copy(ByteStreams.java:167)
at com.google.common.io.ByteStreams.toByteArray(ByteStreams.java:210)
at model.BaseModel.(BaseModel.java:21)
at model.Bert.(Bert.java:10)
at EnsembleDemo._load_Bert(EnsembleDemo.java:56)
at EnsembleDemo.loadModels(EnsembleDemo.java:50)
at EnsembleDemo.(EnsembleDemo.java:21)
at EnsembleDemo.main(EnsembleDemo.java:164)
Exception in thread "main" java.lang.NullPointerException
at model.Bert.predict_single_case(Bert.java:21)
at EnsembleDemo.__pred_with_Bert(EnsembleDemo.java:79)
at EnsembleDemo.doPredictWithModels(EnsembleDemo.java:63)
at EnsembleDemo.main(EnsembleDemo.java:174)

Is this Exception because of the pb file?
how to generate the pb file correctly? i generate it through the code:

frozen_graph = convert_variables_to_constants(sess, input_graph_def, output_names)

i don't know how auther generate it.
and or other problem cause the Exception.