+ Changed ML DownloadType

SkinCanOrg · Jun 9, 2022 · 880c8d0 · 880c8d0
1 parent ced6bf4
commit 880c8d0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/app/src/main/kotlin/io/github/skincanorg/skincan/ui/camera/ScannerActivity.kt b/app/src/main/kotlin/io/github/skincanorg/skincan/ui/camera/ScannerActivity.kt
@@ -127,7 +127,11 @@ class ScannerActivity : AppCompatActivity() {
 
     private fun doPrediction(bitmap: Bitmap) {
         FirebaseModelDownloader.getInstance()
-            .getModel("CancerDetector", DownloadType.LATEST_MODEL, CustomModelDownloadConditions.Builder().build())
+            .getModel(
+                "CancerDetector",
+                DownloadType.LOCAL_MODEL_UPDATE_IN_BACKGROUND,
+                CustomModelDownloadConditions.Builder().build(),
+            )
             .addOnSuccessListener { model ->
                 val modelFile = model?.file ?: return@addOnSuccessListener
                 val interpreter = Interpreter(modelFile)