From 3a0d6e808b399e2b398693751dcd9a4e2da0b00f Mon Sep 17 00:00:00 2001
From: Neeraj Gupta <254676+ua741@users.noreply.github.com>
Date: Tue, 20 Aug 2024 12:54:26 +0530
Subject: [PATCH] [mob] Fix tensor size for clip image

---
 .../src/main/kotlin/io/ente/photos/onnx_dart/OnnxDartPlugin.kt | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mobile/plugins/onnx_dart/android/src/main/kotlin/io/ente/photos/onnx_dart/OnnxDartPlugin.kt b/mobile/plugins/onnx_dart/android/src/main/kotlin/io/ente/photos/onnx_dart/OnnxDartPlugin.kt
index eaf32c21e1..5b84568b2e 100644
--- a/mobile/plugins/onnx_dart/android/src/main/kotlin/io/ente/photos/onnx_dart/OnnxDartPlugin.kt
+++ b/mobile/plugins/onnx_dart/android/src/main/kotlin/io/ente/photos/onnx_dart/OnnxDartPlugin.kt
@@ -178,7 +178,8 @@ class OnnxDartPlugin: FlutterPlugin, MethodCallHandler {
             inputTensorShape = longArrayOf(totalSize, 112, 112, 3)
           }
           ModelType.ClipImageEncoder -> {
-            inputTensorShape = longArrayOf(1, 3, 224, 224)
+            inputTensorShape = longArrayOf(1, 3, 256, 256)
+//            inputTensorShape = longArrayOf(1, 3, 256, 256)
           }
           ModelType.ClipTextEncoder -> {
             inputTensorShape = longArrayOf(1, 77)