Skip to content

Commit 3aa0725

Browse files
committed
update: used name param
1 parent 2a7a4a3 commit 3aa0725

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

ml/src/main/java/com/fpf/smartscansdk/ml/models/providers/embeddings/clip/ClipImageEmbedder.kt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,11 +33,11 @@ class ClipImageEmbedder(
3333

3434
fun isInitialized() = model.isLoaded()
3535

36-
override suspend fun embed(bitmap: Bitmap): FloatArray = withContext(Dispatchers.Default) {
36+
override suspend fun embed(data: Bitmap): FloatArray = withContext(Dispatchers.Default) {
3737
if (!isInitialized()) throw IllegalStateException("Model not initialized")
3838

3939
val inputShape = longArrayOf(ClipConfig.DIM_BATCH_SIZE.toLong(), ClipConfig.DIM_PIXEL_SIZE.toLong(), ClipConfig.IMAGE_SIZE_X.toLong(), ClipConfig.IMAGE_SIZE_Y.toLong())
40-
val imgData: FloatBuffer = preProcess(bitmap)
40+
val imgData: FloatBuffer = preProcess(data)
4141
val inputName = model.getInputNames()?.firstOrNull() ?: throw IllegalStateException("Model inputs not available")
4242
val output = model.run(mapOf(inputName to TensorData.FloatBufferTensor(imgData, inputShape)))
4343
normalizeL2((output.values.first() as Array<FloatArray>)[0])

0 commit comments

Comments
 (0)