import { pipeline } from "@huggingface/transformers";

// Create a text-generation pipeline
const generator = await pipeline(
  "text-generation",
  "HuggingFaceTB/SmolLM2-360M", // You can replace this with other models like "EleutherAI/gpt-neo-125M" or "facebook/opt-125m"
  { device: "webgpu" }
);

// Generate text
const prompts = [
  "Once upon a time",
  "The artificial intelligence"
];

const results = await generator(prompts, {
  max_length: 50,
  num_return_sequences: 1,
  temperature: 0.7,
  top_p: 0.9
});

console.log(results);
// Will output something like:
// [
//   [{
//     "generated_text": "Once upon a time there was a young princess who lived in a castle..."
//   }],
//   [{
//     "generated_text": "The artificial intelligence revolution has transformed the way we..."
//   }]
// ]