last one worked! add new model for text-gen

rework flow into single func for each because output parsing is diff per model

sketch.js

@@ -10,29 +10,56 @@ let PROMPT_INPUT = `Happy people are better than [MASK].` // a field for writing
 let OUTPUT_LIST = [] // a blank array to store the results from the model
 let pField
 
-// RUN MODEL
+// RUN TEXT-GEN MODEL
+
+async function textGenTask(input{
+  console.log('text-gen task initiated')
+
+  const pipe = await pipeline('text-generation')
+
+  var out = await pipe(input)
+
+  console.log(await out)
+  console.log('text-gen task completed')
+
+  // parsing of output
+  await out.forEach(o => {
+    console.log(o)
+    OUTPUT_LIST.push(o.generated_text)
+  })
+
+  console.log(OUTPUT_LIST)
+  console.log('text-gen parsing complete')
+
+  return await OUTPUT_LIST
+  // return await out
+})
+
+// RUN FILL-IN MODEL
 async function fillInTask(input){
   console.log('fill-in task initiated')
 
-  const pipe = await pipeline('fill-mask', 'Xenova/bert-base-uncased');
+  const pipe = await pipeline('fill-mask');
+  //, 'Xenova/bert-base-uncased'
   
   var out = await pipe(input);
 
   console.log(await out) // yields { score, sequence, token, token_str } for each result
+
+  // parsing of output
+  await out.forEach(o => {
+    console.log(o) // yields { score, sequence, token, token_str } for each result
+    OUTPUT_LIST.push(o.sequence) // put only the full sequence in a list
+  })
   
-  // await out.forEach(o => {
-  //   console.log(o) // yields { score, sequence, token, token_str } for each result
-  //   OUTPUT_LIST.push(o.sequence) // put only the full sequence in a list
-  // })
-  
-  // console.log(await OUTPUT_LIST)
+  console.log(await OUTPUT_LIST)
 
   // displayResults(await OUTPUT_LIST)
 
   console.log('fill-in task completed')
 
-  return await out
-  // return await OUTPUT_LIST
+  // return await out
+  return await OUTPUT_LIST
 }
 
 // PROCESS MODEL OUTPUT
@@ -97,9 +124,11 @@ new p5(function (p5){
     PROMPT_INPUT = pField.value() // updates prompt if it's changed
     console.log("latest prompt: ", PROMPT_INPUT)
 
-    let fillIn = await fillInTask(PROMPT_INPUT)
+    // let fillIn = await fillInTask(PROMPT_INPUT)
 
-    let outs = await getOutputs(fillIn)
+    // let outs = await getOutputs(fillIn)
+
+    let outs = await textGenTask(PROMPT_INPUT)
 
     console.log(outs)
   
@@ -109,20 +138,6 @@ new p5(function (p5){
     await outText.html(outs, false)
   }
 
-  // async function makeOutputDisplay(){
-  //   console.log('button pressed')
-  //   let outHead = p5.createElement('h4',"Results:")
-
-  //   let out = await fillInTask() //just model no parsing
-  //   // let out = await getOutputs(fillInTask()) // model and parsing
-
-  //   out = str(await out)
-  //   console.log(out)
-    
-  //   let outText = p5.createP('')
-  //   await outText.html(out)
-  // }
-
 });