fix

Author: Persie

android/src/main/java/com/vladih/computer_vision/flutter_vision/models/Yolo.java

@@ -8,6 +8,7 @@
 import android.util.Log;
 
 import org.tensorflow.lite.Interpreter;
+import org.tensorflow.lite.Tensor;
 import org.tensorflow.lite.gpu.CompatibilityList;
 import org.tensorflow.lite.gpu.GpuDelegate;
 
@@ -57,6 +58,12 @@ public Yolo(Context context,
         this.rotation = rotation;
     }
 
+
+    public Tensor getInputTensor() {
+        if (interpreter == null) return null;
+        return this.interpreter.getInputTensor(0);
+    }
+
     public void initialize_model() throws Exception {
         AssetManager asset_manager = null;
         MappedByteBuffer buffer;