Spaces:
Running
Running
File size: 641 Bytes
7fe3a41 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
const { Conversation } = require('transformers');
async function main() {
// Load the GPT-2 pretrained model
const model = await Conversation.load("gpt2");
// Start conversation
let context = "";
let userMessage = "Hello, chatbot!";
let response = await model.interact(context, userMessage);
console.log("Bot:", response.generated[0].text);
// Continue conversation
context += userMessage + "\n";
userMessage = "Can you tell me about local businesses?";
response = await model.interact(context, userMessage);
console.log("Bot:", response.generated[0].text);
// And so on...
}
main();
|