added code to test local onnx model.

cristianglezm · Oct 21, 2024 · 8c0ed86 · 8c0ed86
1 parent c2eb5f5
commit 8c0ed86
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/store/AIStore/AI.js b/src/store/AIStore/AI.js
@@ -3,10 +3,16 @@ import { pipeline, env } from '@huggingface/transformers';
 env.userBrowserCache = true;
 env.allowLocalModels = false;
 
+//@todo remove this - new onnx local test
+env.allowLocalModels = true;
+env.allowRemoteModels = false;
+env.localModelPath = 'http://localhost/';
+//
+
 const BACKEND = import.meta.env.VITE_APP_DOWNLOAD_URL;
 
 export const isGPUAvailable = () => {
-    return navigator.gpu;
+    return !!navigator.gpu;
 };
 
 export class Captioner{