diff --git a/sample/src/main/java/com/zww/sample/model/AppRepository.kt b/sample/src/main/java/com/zww/sample/model/AppRepository.kt index 44221d65..8ddb98b3 100644 --- a/sample/src/main/java/com/zww/sample/model/AppRepository.kt +++ b/sample/src/main/java/com/zww/sample/model/AppRepository.kt @@ -3,6 +3,7 @@ package com.zww.sample.model import android.app.Application import android.graphics.BitmapFactory import android.os.Environment +import com.blankj.utilcode.util.ToastUtils import com.google.gson.Gson import com.zww.sample.MyNSFWBean import com.zww.sample.ScoreBean @@ -17,7 +18,6 @@ import java.io.File import java.io.FileOutputStream import java.io.InputStream import java.util.* -import kotlin.collections.ArrayList /** * 数据源 这里决定是从缓存获取还是从网络获取 @@ -31,7 +31,7 @@ class AppRepository constructor( val mContext by inject(Application::class.java) val nsfwPath by lazy { - "${Environment.getExternalStorageDirectory().absolutePath}/nsfw.tflite" + "/data/data/com.zww.sample/nsfw.tflite" } //NSFW是否初始化完成 @@ -52,7 +52,7 @@ class AppRepository constructor( val inputStream: InputStream = body.byteStream() //创建一个文件夹 - val file = File(Environment.getExternalStorageDirectory(), "nsfw.tflite") + val file = File(nsfwPath) val outputStream = FileOutputStream(file) val bytes = ByteArray(1024) var len = 0 @@ -82,6 +82,7 @@ class AppRepository constructor( */ override fun initNSFW() { Classifier.Build().context(mContext).isOpenGPU(true).nsfwModuleFilePath(nsfwPath).build() + ToastUtils.showLong("NSFW初始化成功") "NSFW初始化成功".d() isSDKInit = true }