add full text false and max tokens

index.html

@@ -5,7 +5,7 @@
     <meta charset="UTF-8" />
     <link rel="stylesheet" type="text/css" href="style.css" />
     <script src="https://cdnjs.cloudflare.com/ajax/libs/p5.js/1.9.4/p5.js"></script>
-    <script src="https://cdnjs.cloudflare.com/ajax/libs/p5.js/1.9.4/addons/p5.sound.min.js"></script>
+    <!-- <script src="https://cdnjs.cloudflare.com/ajax/libs/p5.js/1.9.4/addons/p5.sound.min.js"></script> -->
     <!-- <meta name="viewport" content="width=device-width, initial-scale=1.0" /> -->
     <title>p5.js Critical AI Prompt Battle</title>
 </head>

sketch.js

@@ -2,7 +2,7 @@ import { pipeline, env } from 'https://cdn.jsdelivr.net/npm/@xenova/transformers
 // import { HfInference } from 'https://cdn.jsdelivr.net/npm/@huggingface/inference@2.7.0/+esm';
 // const inference = new HfInference();
 
-let pipe = await pipeline('text-generation', 'Xenova/bloom-560m');
+let pipe = await pipeline('text-generation', model='Xenova/bloom-560m', return_full_text=False);
 // models('Xenova/gpt2', 'mistralai/Mistral-7B-Instruct-v0.2', 'meta-llama/Meta-Llama-3-8B', 'Xenova/bloom-560m')
 // list of models by task: 'https://huggingface.co/docs/transformers.js/index#supported-tasksmodels'
 
@@ -27,11 +27,11 @@ var PROMPT = `The [BLANK] works as a [FILL] but wishes for [FILL].`
 //     max_tokens: 100
 //   });
 
-let out = await pipe(PREPROMPT + PROMPT)
-// let out = await pipe(PREPROMPT + PROMPT, {
-    // max_new_tokens: 150,
-    // temperature: 0.9
-// });
+// let out = await pipe(PREPROMPT + PROMPT)
+let out = await pipe(PREPROMPT + PROMPT, {
+    max_new_tokens: 250,
+    temperature: 0.9
+});
 console.log(out)
 
 var result = await out[0].generated_text