From 60fb714356ca386f3fc43577373a4d6a3ec0e131 Mon Sep 17 00:00:00 2001
From: Pierre-Yves Nicolas <6371790+pynicolas@users.noreply.github.com>
Date: Sat, 13 Sep 2025 07:12:51 +0200
Subject: [PATCH] Use renamed segmentation model

---
 app/download-tflite.gradle.kts                          | 6 +++---
 app/src/main/java/org/fairscan/app/ImageSegmentation.kt | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/app/download-tflite.gradle.kts b/app/download-tflite.gradle.kts
index d6549de..bc9346e 100644
--- a/app/download-tflite.gradle.kts
+++ b/app/download-tflite.gradle.kts
@@ -1,9 +1,9 @@
 import org.gradle.api.tasks.Copy
 import java.net.URL
 
-val modelVersion = "v0.6.1"
-val modelFileName = "document-segmentation-model.tflite"
-val modelUrl = "https://github.com/pynicolas/document-segmentation-model/releases/download/$modelVersion/$modelFileName"
+val modelVersion = "v1.0.0"
+val modelFileName = "fairscan-segmentation-model.tflite"
+val modelUrl = "https://github.com/pynicolas/fairscan-segmentation-model/releases/download/$modelVersion/$modelFileName"
 
 val downloadedModelPath = layout.buildDirectory.file("downloads/$modelFileName")
 val generatedAssetsDir = layout.buildDirectory.dir("generated/assets")
diff --git a/app/src/main/java/org/fairscan/app/ImageSegmentation.kt b/app/src/main/java/org/fairscan/app/ImageSegmentation.kt
index 862b598..56a3dc4 100644
--- a/app/src/main/java/org/fairscan/app/ImageSegmentation.kt
+++ b/app/src/main/java/org/fairscan/app/ImageSegmentation.kt
@@ -53,7 +53,7 @@ class ImageSegmentationService(private val context: Context) {
 
     fun initialize() {
         interpreter = try {
-            val litertBuffer = FileUtil.loadMappedFile(context, "document-segmentation-model.tflite")
+            val litertBuffer = FileUtil.loadMappedFile(context, "fairscan-segmentation-model.tflite")
             Log.i(TAG, "Loaded LiteRT model")
             val options = Interpreter.Options().apply {
                 numThreads = 2