add model name

index.js

@@ -4,7 +4,8 @@ import { pipeline, env } from 'https://cdn.jsdelivr.net/npm/@xenova/transformers
 
 // import { pipeline } from '@xenova/transformers';
 
-let pipe = await pipeline('text-generation');
+let pipe = await pipeline('text-generation', 'mistralai/Mistral-7B-Instruct-v0.2');
+// models('gpt2', 'mistralai/Mistral-7B-Instruct-v0.2', 'meta-llama/Meta-Llama-3-8B')
 
 // Since we will download the model from the Hugging Face Hub, we can skip the local model check
 // env.allowLocalModels = false;
@@ -29,12 +30,14 @@ var PROMPT = `The [BLANK] works as a [FILL] but wishes for [FILL].`
 let out = await pipe(PREPROMPT + PROMPT);
 console.log(out)
 
-var result = await out.choices[0].message;
+// 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.content
+modelOutput = result
 
 // console.log('huggingface loaded');