Make gradle download the TFLite segmentation model (#8)

2025-06-21 13:01:05 +02:00
parent 7056393f56
commit 0ea4132d37
3 changed files with 42 additions and 1 deletions
--- a/app/src/main/java/org/mydomain/myscan/ImageSegmentation.kt
+++ b/app/src/main/java/org/mydomain/myscan/ImageSegmentation.kt
@@ -53,7 +53,7 @@ class ImageSegmentationService(private val context: Context) {

    fun initialize() {
        interpreter = try {
-            val litertBuffer = FileUtil.loadMappedFile(context, "timm_efficientnet_lite0_quantized.tflite")
+            val litertBuffer = FileUtil.loadMappedFile(context, "document-segmentation-model.tflite")
            Log.i(TAG, "Loaded LiteRT model")
            val options = Interpreter.Options().apply {
                numThreads = 2