update bindings

Vinayak006 · Aug 13, 2021 · e5601a8 · e5601a8
1 parent a063b1c
commit e5601a8
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 3 deletions.
diff --git a/lib/src/task/bindings/dlib.dart b/lib/src/task/bindings/dlib.dart
@@ -29,7 +29,7 @@ String get binaryName {
 // ignore: missing_return
 DynamicLibrary tflitelib = () {
   if (Platform.isAndroid) {
-    return DynamicLibrary.open('libtensorflowlite_c.so');
+    return DynamicLibrary.open('libtask_text_jni.so');
   } else if (Platform.isIOS) {
     return DynamicLibrary.process();
   } else {

diff --git a/lib/src/task/text/nl_classifier/nl_classifier.dart b/lib/src/task/text/nl_classifier/nl_classifier.dart
@@ -13,7 +13,8 @@ class NLClassifier {
 
   NLClassifier._(this._classifier);
 
-  factory NLClassifier._create(String modelPath, NLClassifierOptions options) {
+  // TODO: create convenience constructors
+  factory NLClassifier.create(String modelPath, NLClassifierOptions options) {
     final classiferPtr =
         NLClassifierFromFileAndOptions(modelPath.toNativeUtf8(), options.base);
     return NLClassifier._(classiferPtr);

diff --git a/lib/tflite_flutter_helper.dart b/lib/tflite_flutter_helper.dart
@@ -39,3 +39,5 @@ export 'src/tensorbuffer/tensorbufferuint8.dart';
 export 'src/audio/recorder_stream.dart';
 export 'src/audio/sound_stream.dart';
 export 'src/audio/tensor_audio.dart';
+export 'src/task/text/nl_classifier/nl_classifier.dart';
+export 'src/task/text/nl_classifier/nl_classifier_options.dart';
diff --git a/pubspec.yaml b/pubspec.yaml
@@ -12,7 +12,8 @@ dependencies:
   meta: ^1.1.8
   quiver: ^3.0.1
   path_provider: ^2.0.1
-  tflite_flutter: ^0.9.0
+  tflite_flutter:
+    path: ../tflite_flutter_plugin
   image: ^3.0.2
   tuple: ^2.0.0
   ffi: ^1.0.0