switch model add hyperparams

README.md

@@ -6,8 +6,9 @@ colorTo: blue
 sdk: static
 pinned: false
 models:
-- Xenova/detr-resnet-50
+# - Xenova/detr-resnet-50
 - Xenova/gpt2
+- Xenova/LaMini-Flan-T5-783M
 # - mistralai/Mistral-7B-Instruct-v0.2
 # - meta-llama/Meta-Llama-3-8B
 ---

sketch.js

@@ -2,16 +2,16 @@ 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();
 
-// import { pipeline } from '@xenova/transformers';
-
-let pipe = await pipeline('text-generation', 'Xenova/gpt2');
+let pipe = await pipeline('text-generation', 'Xenova/LaMini-Flan-T5-783M');
 // models('Xenova/gpt2', 'mistralai/Mistral-7B-Instruct-v0.2', 'meta-llama/Meta-Llama-3-8B')
+// list of models by task: 'https://huggingface.co/docs/transformers.js/index#supported-tasksmodels'
+
 
 // Since we will download the model from the Hugging Face Hub, we can skip the local model check
-// env.allowLocalModels = false;
+env.allowLocalModels = false;
 
-let promptButton, buttonButton, promptInput, maskInputA, maskInputB, maskInputC, modOutput, modelOutput
-// const detector = await pipeline('text-generation', 'meta-llama/Meta-Llama-3-8B');
+let promptButton, buttonButton, promptInput, maskInputA, maskInputB, maskInputC, modOutput
+// const detector = await pipeline('text-generation', 'meta-llama/Meta-Llama-3-8B', 'Xenova/LaMini-Flan-T5-783M');
 
 var inputArray = ["Brit", "Israeli", "German", "Palestinian"]
 
@@ -27,18 +27,13 @@ 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: 100,
+    temperature: 0.9
+});
 console.log(out)
 
-// var result = await out.choices[0].message;
 var result = await out.generated_text
-// console.log("role: ", result.role, "content: ", result.content);
-
-//sends the text to a global var (not best way cant figure out better)
-// window.modelOutput = result.content;
-// modelOutput = result.content
-modelOutput = result
-
 // console.log('huggingface loaded');
   
   
@@ -145,14 +140,14 @@ new p5(function(p5){
         // p5.background(200)
         // p5.textSize(20)
         // p5.textAlign(p5.CENTER,p5.CENTER)
-        // let promptButton = p5.createButton("GO").position(0, 340);
-        // promptButton.position(0, 340);
-        // promptButton.elt.style.fontSize = "15px";
+        let promptButton = p5.createButton("GO").position(0, 340);
+        promptButton.position(0, 340);
+        promptButton.elt.style.fontSize = "15px";
         
     }
     
     p5.draw = function(){
-        pass
+        //
     }
 
     window.onload = function(){
@@ -189,7 +184,7 @@ new p5(function(p5){
         modOutput = p5.createElement("p", "Results:");
         modOutput.position(0, 380);
             setTimeout(() => {
-                modOutput.html(modelOutput)
+                modOutput.html(result)
         }, 2000);
 
     }