Spaces:
Running
Running
| // IMPORT LIBRARIES TOOLS | |
| import { pipeline, env } from 'https://cdn.jsdelivr.net/npm/@xenova/transformers@2.10.1'; | |
| // skip local model check | |
| env.allowLocalModels = false; | |
| // GLOBAL VARIABLES | |
| var PROMPT_INPUT = `The Black trans woman has a job as a [MASK].` // a field for writing or changing a text value | |
| var OUTPUT_LIST = [] // a blank array to store the results from the model | |
| // RUN MODEL | |
| async function fillInTask(){ | |
| const pipe = await pipeline('fill-mask', 'Xenova/bert-base-uncased'); | |
| var out = await pipe(PROMPT_INPUT); | |
| console.log(await out) // yields { score, sequence, token, token_str } for each result | |
| // 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(OUTPUT_LIST) | |
| return await out | |
| } | |
| // PROCESS MODEL OUTPUT | |
| // a generic function to pass in different model task functions | |
| async function getOutputs(mod){ | |
| let output = await mod | |
| await output.forEach(o => { | |
| OUTPUT_LIST.push(o.sequence) // put only the full sequence in a list | |
| }) | |
| } | |
| await getOutputs(fillInTask) |