bfix loadFileOnDevice

Vinayak006 · Aug 19, 2021 · 3fa5ecd · 3fa5ecd
1 parent 8827816
commit 3fa5ecd
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/lib/src/common/file_util.dart b/lib/src/common/file_util.dart
@@ -53,12 +53,15 @@ class FileUtil {
     return list;
   }
 
-  /// Loads a file from project assets on the device
-  static Future<File> loadFileOnDevice(String fileAssetLocation) async {
+  /// Loads a file from project assets at [fileAssetPath] on the device
+  ///
+  /// Eg. 'assets/models/my_model.tflite'
+  static Future<File> loadFileOnDevice(String fileAssetPath) async {
     final appDir = await getTemporaryDirectory();
     final appPath = appDir.path;
-    final fileOnDevice = File('$appPath/$fileAssetLocation');
-    final rawAssetFile = await rootBundle.load('$fileAssetLocation');
+    final fileName = fileAssetPath.split('/').last;
+    final fileOnDevice = File('$appPath/$fileName');
+    final rawAssetFile = await rootBundle.load('$fileAssetPath');
     final rawBytes = rawAssetFile.buffer.asUint8List();
     await fileOnDevice.writeAsBytes(rawBytes, flush: true);
     return fileOnDevice;