try new model chat template

sketch.js

@@ -1,13 +1,13 @@
 
 // IMPORT LIBRARIES TOOLS
-import { pipeline, env } from 'https://cdn.jsdelivr.net/npm/@xenova/[email protected]';
+import { pipeline, env, AutoTokenizer } from 'https://cdn.jsdelivr.net/npm/@xenova/[email protected]';
 // AutoTokenizer
 
 // skip local model check
 env.allowLocalModels = false;
 
 // GLOBAL VARIABLES
-let PROMPT_INPUT = `The woman has a job as a...` // a field for writing or changing a text value
+let PROMPT_INPUT = `The woman has a job as a [MASK] and likes to ...` // a field for writing or changing a text value
 let pField
 let PREPROMPT = `You're a friendly pirate. Please complete the phrase and fill in any [MASK].`
 
@@ -16,20 +16,26 @@ let PREPROMPT = `You're a friendly pirate. Please complete the phrase and fill i
 async function textGenTask(input){
   console.log('text-gen task initiated')
 
-  let MODEL = 'Xenova/OpenELM-270M-Instruct'
-  // const = modelsList = ['Xenova/LaMini-Cerebras-256M', 'Xenova/TinyLlama-1.1B-Chat-v1.0']
+  let MODEL = 'Xenova/stablelm-2-zephyr-1_6b'
+  // const = modelsList = ['Xenova/stablelm-2-zephyr-1_6b', 'Xenova/LaMini-Cerebras-256M', 'Xenova/TinyLlama-1.1B-Chat-v1.0']
 
   const pipe = await pipeline('text-generation', MODEL)
-  // let tokenizer = AutoTokenizer.from_pretrained(MODEL)
+  let tokenizer = AutoTokenizer.from_pretrained(MODEL)
 
-  // let messages = [
+  // const messages = [
   //     {"role": "system", "content": PREPROMPT},
   //     {"role": "user", "content": input}
   //   ]
 
-  // const prompt = pipe.tokenizer.apply_chat_template(messages, {
-  //   tokenize: false, add_generation_prompt: false,
-  // });
+  const messages = [
+    { "role": "system", "content": "You are a helpful assistant." },
+    { "role": "user", "content": input },
+  ]
+
+  const prompt = pipe.tokenizer.apply_chat_template(messages, {
+    tokenize: false, 
+    add_generation_prompt: false,
+  });
 
   // run text through model, setting hyperparameters
   var out = await pipe(input, {