From c125b4d0b8a39f6ef857c1019b8e60a21631933e Mon Sep 17 00:00:00 2001 From: Yelin Jeong Date: Fri, 20 Sep 2024 11:19:01 +0900 Subject: [PATCH] Data/Model: Supports multiple framework This patch removes hard-coded tensorflow-lite to supports multiple framework. Signed-off-by: Yelin Jeong --- .../java/ai/nnstreamer/ml/inference/offloading/data/Model.kt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/data/Model.kt b/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/data/Model.kt index e56e824..b468d2d 100644 --- a/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/data/Model.kt +++ b/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/data/Model.kt @@ -179,7 +179,7 @@ data class Model( val filter = "other/tensors,format=${inFormat}${inTensors}${inDims}${inTypes},framerate=0/1 ! " + - "tensor_filter framework=tensorflow-lite model=${modelPaths} ! " + + "tensor_filter framework=${framework} model=${modelPaths} ! " + "other/tensors,format=${outFormat}${outTensors}${outDims}${outTypes},framerate=0/1" return filter