修复demo bug

pull/34/head
赵文文 5 years ago
parent b0f57acad2
commit 10b2a1f300
  1. 7
      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
}

Loading…
Cancel
Save