Update README.md

README.md

@@ -12,17 +12,9 @@ INT8 ONNX version of [Felladrin/TinyMistral-248M-SFT-v4](https://huggingface.co/
 import { pipeline } from '@xenova/transformers';
 
 const prompt = `<|im_start|>user
-Question: What is a dog?<|im_end|>
-<|im_start|>assistant
-A dog is a four-legged, domesticated animal that is a member of the class Mammalia, which includes all mammals. Dogs are known for their loyalty, playfulness, and ability to be trained for various tasks. They are also used for hunting, herding, and as service animals.<|im_end|>
-<|im_start|>user
-Question: What is the capital of France?<|im_end|>
-<|im_start|>assistant
-The capital of France is Paris. Paris is located in the north-central region of France and is known for its famous landmarks, such as the Eiffel Tower, the Louvre Museum, and Notre-Dame Cathedral.<|im_end|>
-<|im_start|>user
-Question: What is the color of an apple?<|im_end|>
+Question: What are some potential applications for quantum computing?<|im_end|>
 <|im_start|>assistant`;
-const generator = await pipeline('text-generation', 'Felladrin/onnx-TinyMistral-248M-SFT-v4', { quantized: false });
-const output = await generator(prompt, { max_new_tokens: 100, penalty_alpha: 0.5, top_k: 5 });
-console.log(output[0].generated_text); // "The color of a apple is reddish-brown, with a creamy, floral texture that combines the bright and vibrant colors of the apples and oranges. It is a popular dessert in many countries, and is often associated with festive gatherings and celebrations.<|im_end|>"
+const generator = await pipeline('text-generation', 'Felladrin/onnx-TinyMistral-248M-SFT-v4');
+const output = await generator(prompt, { add_special_tokens: true, max_new_tokens: 128, repetition_penalty: 1.079 });
+console.log(output[0].generated_text); // "Quantum computers have the potential to revolutionize the way we communicate and access information. They can be used to perform complex calculations, such as storing large amounts of data in a secure location, or even allowing users to access data from anywhere in the world. Quantum computers also have many applications in fields such as medicine, finance, and healthcare.<|im_end|>"
 ```