How I can load and predict With Flask?

[DevanshMistry890]

Hello, I am wanting to make a front end for the ABSA project, so how can I dump/save the model so later I can load it while runtime in flask? I tried to save checkpoints but it is not working. Any help will be appreciated.

[yangheng95]

Generally, the checkpoints will be saved if you set checkpoint_save_mode in a trainer interface. What exception did you get while saving a checkpoint?

[yangheng95]

Did run the code on Colab?

[DevanshMistry890]

I am wanting to work with a flask, so tested it locally.

[DevanshMistry890]

Protobuf also has a native os version so, after installing it above error is solved. Now checkpoints are working. But how can I load my model in the flask, I tried to call using torch.load but not working.

[yangheng95]

Could you please paste your code snippet and the console oupout which is needed to locate prolem?

[DevanshMistry890]

how I can serve this model using flask rest API?