From 9a29c1d4f33dfb7d9b1f91ab629a701bab5b8016 Mon Sep 17 00:00:00 2001 From: navanchauhan Date: Mon, 28 Jun 2021 00:50:36 +0530 Subject: fixed code snippet --- ...2021-06-27-Crude-ML-AI-Powered-Chatbot-Swift.md | 24 ++++++++++++++++------ 1 file changed, 18 insertions(+), 6 deletions(-) (limited to 'Content') diff --git a/Content/posts/2021-06-27-Crude-ML-AI-Powered-Chatbot-Swift.md b/Content/posts/2021-06-27-Crude-ML-AI-Powered-Chatbot-Swift.md index 0dcabec..804ad97 100644 --- a/Content/posts/2021-06-27-Crude-ML-AI-Powered-Chatbot-Swift.md +++ b/Content/posts/2021-06-27-Crude-ML-AI-Powered-Chatbot-Swift.md @@ -105,14 +105,26 @@ Otherwise, it calls the custom action. ![Screenshot](/assets/posts/swift-chatbot/carbon-4.png) ```swift -let defaultResponses = [ - "greetings": "Hello", - "banter": "no, plix no" +let sampleMessages = [ + "Hey there, how is it going", + "hello, there", + "Who let the dogs out", + "can you tell me about the compound Geraniin", + "what do you know about the compound Ibuprofen", + "please, tell me more about the compound", + "please, tell me more about the molecule dihydrogen-monoxide" ] -let customActions = [ - "deez-drug": customAction -] +for sampleMessage in sampleMessages { + User.message = sampleMessage + let prediction = intentPredictor.predictedLabel(for: sampleMessage) + + if (defaultResponses[prediction!] != nil) { + print(defaultResponses[prediction!]!) + } else if (customActions[prediction!] != nil) { + print(customActions[prediction!]!()) + } +} ``` ![Output](/assets/posts/swift-chatbot/output.png) -- cgit v1.2.3