Spaces:
Running
Running
sarahciston
commited on
reformat parameter, change prompt, parse result
Browse files
index.js
CHANGED
@@ -5,16 +5,13 @@ env.allowLocalModels = false;
|
|
5 |
|
6 |
const unmasker = await pipeline('fill-mask', 'Xenova/bert-base-uncased');
|
7 |
|
8 |
-
var out = await unmasker("The woman has a job as a [MASK] and hates it.",
|
9 |
-
parameters: {
|
10 |
-
top_k: 7
|
11 |
-
})
|
12 |
|
13 |
var outputList = []
|
14 |
|
15 |
out.forEach(o => {
|
16 |
-
console.log(o)
|
17 |
-
outputList.push(o.sequence)
|
18 |
})
|
19 |
|
20 |
console.log(outputList)
|
|
|
5 |
|
6 |
const unmasker = await pipeline('fill-mask', 'Xenova/bert-base-uncased');
|
7 |
|
8 |
+
var out = await unmasker("The woman has a job as a [MASK] and hates it.", {top_k: 7});
|
|
|
|
|
|
|
9 |
|
10 |
var outputList = []
|
11 |
|
12 |
out.forEach(o => {
|
13 |
+
console.log(o) // yields { score, sequence, token, token_str } for each result
|
14 |
+
outputList.push(o.sequence) // put only the full sequence in a list
|
15 |
})
|
16 |
|
17 |
console.log(outputList)
|