From 307e666c27f3e1632ff6090469e691a7f59f168f Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Sat, 9 Sep 2023 16:42:16 +0300
Subject: [PATCH] onnx runtime to web

---
 src/classify.mjs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/classify.mjs b/src/classify.mjs
index 635b2642e..c4be593b4 100644
--- a/src/classify.mjs
+++ b/src/classify.mjs
@@ -1,6 +1,8 @@
-import { pipeline, TextClassificationPipeline } from 'sillytavern-transformers';
+import { pipeline, TextClassificationPipeline, env } from 'sillytavern-transformers';
 import path from 'path';
 
+env.backends.onnx = 'onnxruntime-web';
+
 class PipelineAccessor {
     /**
      * @type {TextClassificationPipeline}