PeterPinetree commited on
Commit
d407658
·
verified ·
1 Parent(s): 9783073

Update index.html

Browse files
Files changed (1) hide show
  1. index.html +2 -1
index.html CHANGED
@@ -94,6 +94,7 @@
94
 
95
  tf.env.useBrowserCache = true;
96
  tf.env.allowLocalModels = true; // <-- REQUIRED for local folder paths
 
97
 
98
  const $ = s => document.querySelector(s);
99
  const modelSel = $('#model');
@@ -115,7 +116,7 @@
115
  try {
116
  if (modelId === 'local:gpt2') {
117
  // Note: no double slashes, no /resolve/main – just your folder.
118
- tokenizer = await tf.AutoTokenizer.from_pretrained('./assets/gpt2');
119
  } else {
120
  tokenizer = await tf.AutoTokenizer.from_pretrained(modelId);
121
  }
 
94
 
95
  tf.env.useBrowserCache = true;
96
  tf.env.allowLocalModels = true; // <-- REQUIRED for local folder paths
97
+ tf.env.localModelPath = '/';
98
 
99
  const $ = s => document.querySelector(s);
100
  const modelSel = $('#model');
 
116
  try {
117
  if (modelId === 'local:gpt2') {
118
  // Note: no double slashes, no /resolve/main – just your folder.
119
+ tokenizer = await tf.AutoTokenizer.from_pretrained('assets/gpt2');
120
  } else {
121
  tokenizer = await tf.AutoTokenizer.from_pretrained(modelId);
122
  }