Skip to content

Instantly share code, notes, and snippets.

@frankfliu
frankfliu / LoadModelFromJar.java
Last active January 27, 2022 09:57
Load PyTorch model for jar file
import ai.djl.Application;
import ai.djl.ModelException;
import ai.djl.inference.Predictor;
import ai.djl.modality.Classifications;
import ai.djl.modality.cv.Image;
import ai.djl.modality.cv.ImageFactory;
import ai.djl.repository.zoo.Criteria;
import ai.djl.repository.zoo.ZooModel;
import ai.djl.translate.TranslateException;