From 2804004525c65df99cd8a563cfd316e796618ebd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tarek=20Ziad=C3=A9?= <tarek@ziade.org>
Date: Wed, 10 Apr 2024 10:06:19 +0200
Subject: [PATCH 1/1] patched
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Tarek Ziadé <tarek@ziade.org>
---
 package.json         |  2 +-
 src/backends/onnx.js |  8 ++++++--
 tsconfig.json        | 13 +++++++++++++
 webpack.config.js    |  4 ++++
 4 files changed, 24 insertions(+), 3 deletions(-)
 create mode 100644 tsconfig.json

diff --git a/package.json b/package.json
index 446fd5f..24825d6 100644
--- a/package.json
+++ b/package.json
@@ -6,7 +6,7 @@
   "types": "./types/transformers.d.ts",
   "type": "module",
   "scripts": {
-    "typegen": "tsc ./src/transformers.js --allowJs --declaration --emitDeclarationOnly --declarationMap --outDir types",
+    "typegen": "tsc",
     "dev": "webpack serve --no-client-overlay",
     "build": "webpack && npm run typegen",
     "generate-tests": "python -m tests.generate_tests",
diff --git a/src/backends/onnx.js b/src/backends/onnx.js
index 0bee3dc..030eff7 100644
--- a/src/backends/onnx.js
+++ b/src/backends/onnx.js
@@ -18,8 +18,12 @@
 
 // NOTE: Import order matters here. We need to import `onnxruntime-node` before `onnxruntime-web`.
 // In either case, we select the default export if it exists, otherwise we use the named export.
-import * as ONNX_NODE from 'onnxruntime-node';
-import * as ONNX_WEB from 'onnxruntime-web';
+//import * as ONNX_NODE from 'onnxruntime-node';
+//import * as ONNX_WEB from 'onnxruntime-web';
+
+import 'onnxruntime-web';
+const ONNX_WEB = ort;
+
 
 /** @type {import('onnxruntime-web')} The ONNX runtime module. */
 export let ONNX;
diff --git a/tsconfig.json b/tsconfig.json
new file mode 100644
index 0000000..8892b18
--- /dev/null
+++ b/tsconfig.json
@@ -0,0 +1,13 @@
+{
+  "files": [
+    "./src/transformers.js"
+  ],
+  "compilerOptions": {
+    "allowJs": true,
+    "declaration": true,
+    "emitDeclarationOnly": true,
+    "declarationMap": true,
+    "outDir": "types",
+    "types": [] // Or specify only the types you need
+  }
+}
diff --git a/webpack.config.js b/webpack.config.js
index c958b45..8b6b98b 100644
--- a/webpack.config.js
+++ b/webpack.config.js
@@ -51,4 +51,8 @@ export default {
     experiments: {
         outputModule: true,
     },
+    externals: {
+        'onnxruntime-node': 'NOT_USED',
+        'onnxruntime-web': 'chrome://global/content/ml/ort-dev.js',
+    },
 };
-- 
2.43.0