[js/rn] Support load external data (#20090)

Support load external data by passing local model path
microsoft · Apr 5, 2024 · 6abfb6b · 6abfb6b
1 parent f61cca1
commit 6abfb6b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/js/react_native/android/src/main/java/ai/onnxruntime/reactnative/OnnxruntimeModule.java b/js/react_native/android/src/main/java/ai/onnxruntime/reactnative/OnnxruntimeModule.java
@@ -199,6 +199,12 @@ private WritableMap loadModelImpl(String uri, byte[] modelData, ReadableMap opti
     if (modelData != null && modelData.length > 0) {
       // load model via model data array
       ortSession = ortEnvironment.createSession(modelData, sessionOptions);
+    } else if (uri.startsWith("file://") || uri.startsWith("/")) {
+      // load model from local
+      if (uri.startsWith("file://")) {
+        uri = uri.substring(7);
+      }
+      ortSession = ortEnvironment.createSession(uri, sessionOptions);
     } else {
       // load model via model path string uri
       InputStream modelStream =