Data representation in Bar

def bar_plot(dir):
    x=len(os.listdir(dir+'/ok'))
    y=len(os.listdir(dir+'/nok'))
    category = ['ok', 'nok']
    count = [x, y]
    plot = plt.bar(category,count) 
    plot[0].set_color('green')
    plt.title('Plot of number of values for each category')
    plt.show()

print('Training images:\n')
print('OK:', len(os.listdir(train_dir+'/ok')))
print('NOK:', len(os.listdir(train_dir+'/nok')))
print('Total Training images:', len(os.listdir(train_dir+'/ok')) + len(os.listdir(train_dir+'/nok')))
print('*'*49)
bar_plot(train_dir)


print('Testing images:\n')
print('OK:', len(os.listdir(test_dir+'/ok')))
print('NOK:', len(os.listdir(test_dir+'/nok')))
print('Total Testing images:', len(os.listdir(test_dir+'/ok')) + len(os.listdir(test_dir+'/nok')))
print('*'*49)
bar_plot(test_dir)

print('Validation images:\n')
print('OK:', len(os.listdir(val_dir+'/ok')))
print('NOK:', len(os.listdir(val_dir+'/nok')))
print('Total Testing images:', len(os.listdir(val_dir+'/ok')) + len(os.listdir(val_dir+'/nok')))
print('*'*49)
bar_plot(val_dir)