sarahciston commited on
Commit
04273ac
·
verified ·
1 Parent(s): a853310

try `Xenova/t5-small` model for text-gen

Browse files
Files changed (1) hide show
  1. sketch.js +2 -2
sketch.js CHANGED
@@ -6,7 +6,7 @@ import { pipeline, env } from 'https://cdn.jsdelivr.net/npm/@xenova/transformers
6
  env.allowLocalModels = false;
7
 
8
  // GLOBAL VARIABLES
9
- let PROMPT_INPUT = `Happy people are better than [MASK].` // a field for writing or changing a text value
10
  let pField
11
 
12
  // RUN TEXT-GEN MODEL
@@ -14,7 +14,7 @@ let pField
14
  async function textGenTask(input){
15
  console.log('text-gen task initiated')
16
 
17
- const pipe = await pipeline('text-generation', 'HuggingFaceH4/zephyr-7b-beta')
18
 
19
  var out = await pipe(input)
20
 
 
6
  env.allowLocalModels = false;
7
 
8
  // GLOBAL VARIABLES
9
+ let PROMPT_INPUT = `Happy people like to talk about [MASK] and...` // a field for writing or changing a text value
10
  let pField
11
 
12
  // RUN TEXT-GEN MODEL
 
14
  async function textGenTask(input){
15
  console.log('text-gen task initiated')
16
 
17
+ const pipe = await pipeline('text-generation', 'Xenova/t5-small')
18
 
19
  var out = await pipe(input)
20