change options syntax for model settings

README.md

@@ -6,15 +6,16 @@ colorTo: blue
 sdk: static
 pinned: false
 models:
+- openai-community/gpt2
 # - meta-llama/Meta-Llama-3-70B-Instruct
 # - Xenova/detr-resnet-50
-- Xenova/gpt2
+# - Xenova/gpt2
 # - Xenova/bloom-560m
-- Xenova/distilgpt2
+# - Xenova/distilgpt2
 # - Xenova/gpt-3.5-turbo
 # - 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

@@ -207,11 +207,16 @@ async function runModel(PREPROMPT, PROMPT){
     // return modelResult
 
     // pipeline/transformers version TEST
-    let pipe = await pipeline('text-generation');
+    let pipe = await pipeline('text-generation', 'openai-community/gpt2');
 
     // , 'meta-llama/Meta-Llama-3-70B-Instruct'
+    // , 'openai-community/gpt2'
 
-    out = pipe((PREPROMPT, PROMPT), return_full_text = false)
+    out = await pipe((PREPROMPT, PROMPT), {
+        max_tokens: 150,
+        num_return_sequences: 2, 
+        return_full_text: false
+    })
     
     console.log(out)