try parsing messages again

sketch.js

@@ -19,16 +19,16 @@ async function textGenTask(input){
 
   // const = modelsList = ['Xenova/LaMini-Cerebras-256M', ]
 
-  // let messages = [
-  //     {"role": "system", "content": PREPROMPT},
-  //     {"role": "user", "content": input}
-  //   ]
+  let messages = [
+      {"role": "system", "content": PREPROMPT},
+      {"role": "user", "content": input}
+    ]
 
   // const prompt = pipe.tokenizer.apply_chat_template(messages, {
-  //   tokenize: false, add_generation_prompt: true,
+  //   tokenize: false, add_generation_prompt: false,
   // });
 
-  var out = await pipe(input, {
+  var out = await pipe(messages, {
     max_new_tokens: 256,
     temperature: 0.7,
     do_sample: true,
@@ -41,16 +41,16 @@ async function textGenTask(input){
   let OUTPUT_LIST = [] // a blank array to store the results from the model
 
   // parsing of output
-  // await out.forEach(o => {
-  //   console.log(o)
-  //   OUTPUT_LIST.push(o.generated_text)
-  // })
-
-  await out.choices.forEach(o => {
+  await out.forEach(o => {
     console.log(o)
-    OUTPUT_LIST.push(o.message.content)
+    OUTPUT_LIST.push(o.generated_text)
   })
 
+  // await out.choices.forEach(o => {
+  //   console.log(o)
+  //   OUTPUT_LIST.push(o.message.content)
+  // })
+
   console.log(OUTPUT_LIST)
   console.log('text-gen parsing complete')