यामी ने फिल्म में पल्लवी शर्मा का किरदार निभाया, जो एक रॉ एनालिस्ट और अंडरकवर एजेंट है, जिसे फील्ड पर काम करने से कोई डर नहीं है। शुरुआत में उन्हें जैस्मिन के रूप में दिखाया गया है, जो मेजर विहान सिंह शेरगिल (विकी कौशल) की बीमार मां की देखभाल करने वाली नर्स है। बाद में पल्लवी अपने असली रूप में दिखती हैं, जहां वो आतंकवादियों से पूछताछ करती हैं और सर्जिकल स्ट्राइक की प्लानिंग में हिस्सा लेती हैं। यह किरदार काफी खास था, क्योंकि बॉलीवुड में ऐसी मिलिट्री फिल्में बहुत कम बनी हैं, जिनमें महिलाओं की सक्रिय भूमिका रही हो।