Spaces:
Running
Running
update results interface section, try new hyperparams
Browse files
sketch.js
CHANGED
|
@@ -6,9 +6,10 @@ import { pipeline, env } from 'https://cdn.jsdelivr.net/npm/@xenova/transformers
|
|
| 6 |
env.allowLocalModels = false;
|
| 7 |
|
| 8 |
// GLOBAL VARIABLES
|
|
|
|
| 9 |
let PROMPT_INPUT = `The woman has a job as a [MASK] but wishes to...` // a field for writing or changing a text value
|
| 10 |
let pField
|
| 11 |
-
let
|
| 12 |
|
| 13 |
// RUN TEXT-GEN MODEL
|
| 14 |
|
|
@@ -26,7 +27,7 @@ async function textGenTask(pre, prompt){
|
|
| 26 |
const pipe = await pipeline('text-generation', MODEL)
|
| 27 |
|
| 28 |
// RUN INPUT THROUGH MODEL,
|
| 29 |
-
var out = await pipe(INPUT, { max_new_tokens: 60 })
|
| 30 |
// setting hyperparameters
|
| 31 |
// max_new_tokens: 256, top_k: 50, temperature: 0.7, do_sample: true,
|
| 32 |
|
|
@@ -134,17 +135,17 @@ new p5(function (p5){
|
|
| 134 |
submitButton.size(170)
|
| 135 |
submitButton.class('submit')
|
| 136 |
submitButton.mousePressed(displayResults)
|
|
|
|
|
|
|
| 137 |
let outHeader = p5.createElement('h3',"Results")
|
|
|
|
| 138 |
}
|
| 139 |
|
| 140 |
async function displayResults(){
|
| 141 |
console.log('submitButton pressed')
|
| 142 |
|
| 143 |
-
//
|
| 144 |
-
|
| 145 |
-
if (outText == null) {
|
| 146 |
-
p5.createP('...').id('results')
|
| 147 |
-
}
|
| 148 |
|
| 149 |
PROMPT_INPUT = pField.value() // grab update to the prompt if it's been changed
|
| 150 |
console.log("latest prompt: ", PROMPT_INPUT)
|
|
@@ -155,7 +156,7 @@ new p5(function (p5){
|
|
| 155 |
console.log(outs)
|
| 156 |
|
| 157 |
// insert the model outputs into the paragraph
|
| 158 |
-
await outText.html(outs,
|
| 159 |
}
|
| 160 |
|
| 161 |
});
|
|
|
|
| 6 |
env.allowLocalModels = false;
|
| 7 |
|
| 8 |
// GLOBAL VARIABLES
|
| 9 |
+
// let PREPROMPT = `Please complete the phrase and fill in any [MASK]: `
|
| 10 |
let PROMPT_INPUT = `The woman has a job as a [MASK] but wishes to...` // a field for writing or changing a text value
|
| 11 |
let pField
|
| 12 |
+
let outText
|
| 13 |
|
| 14 |
// RUN TEXT-GEN MODEL
|
| 15 |
|
|
|
|
| 27 |
const pipe = await pipeline('text-generation', MODEL)
|
| 28 |
|
| 29 |
// RUN INPUT THROUGH MODEL,
|
| 30 |
+
var out = await pipe(INPUT, { max_new_tokens: 60, top_k: 90, num_return_sequences: 3 })
|
| 31 |
// setting hyperparameters
|
| 32 |
// max_new_tokens: 256, top_k: 50, temperature: 0.7, do_sample: true,
|
| 33 |
|
|
|
|
| 135 |
submitButton.size(170)
|
| 136 |
submitButton.class('submit')
|
| 137 |
submitButton.mousePressed(displayResults)
|
| 138 |
+
|
| 139 |
+
// also make results placeholder
|
| 140 |
let outHeader = p5.createElement('h3',"Results")
|
| 141 |
+
outText = p5.createP('').id('results')
|
| 142 |
}
|
| 143 |
|
| 144 |
async function displayResults(){
|
| 145 |
console.log('submitButton pressed')
|
| 146 |
|
| 147 |
+
// insert waiting dots into results space of interface
|
| 148 |
+
outText.html('...', false)
|
|
|
|
|
|
|
|
|
|
| 149 |
|
| 150 |
PROMPT_INPUT = pField.value() // grab update to the prompt if it's been changed
|
| 151 |
console.log("latest prompt: ", PROMPT_INPUT)
|
|
|
|
| 156 |
console.log(outs)
|
| 157 |
|
| 158 |
// insert the model outputs into the paragraph
|
| 159 |
+
await outText.html(outs, false) // false replaces text instead of appends??
|
| 160 |
}
|
| 161 |
|
| 162 |
});
|