switch models, add consolelogs, result array[0]

README.md

@@ -8,8 +8,10 @@ pinned: false
 models:
 # - Xenova/detr-resnet-50
 - Xenova/gpt2
+- Xenova/bloom-560m
+# - Xenova/llama-68m
 # - Xenova/LaMini-Flan-T5-783M
-- mistralai/Mistral-7B-Instruct-v0.2
+# - mistralai/Mistral-7B-Instruct-v0.2
 # - meta-llama/Meta-Llama-3-8B
 ---
 

sketch.js

@@ -2,8 +2,8 @@ import { pipeline, env } from 'https://cdn.jsdelivr.net/npm/@xenova/transformers
 // import { HfInference } from 'https://cdn.jsdelivr.net/npm/@huggingface/[email protected]/+esm';
 // const inference = new HfInference();
 
-let pipe = await pipeline('text-generation');
-// models('Xenova/gpt2', 'mistralai/Mistral-7B-Instruct-v0.2', 'meta-llama/Meta-Llama-3-8B')
+let pipe = await pipeline('text-generation', 'Xenova/bloom-560m');
+// 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,14 +27,15 @@ var PROMPT = `The [BLANK] works as a [FILL] but wishes for [FILL].`
 //     max_tokens: 100
 //   });
 
-let out = await pipe(PREPROMPT + PROMPT, {
-    max_new_tokens: 100,
-    temperature: 0.9
-});
+let out = await pipe(PREPROMPT + PROMPT)
+// let out = await pipe(PREPROMPT + PROMPT, {
+    // max_new_tokens: 150,
+    // temperature: 0.9
+// });
 console.log(out)
 
-var result = await out.generated_text
-// console.log('huggingface loaded');
+var result = await out[0].generated_text
+console.log(result);