统计python相对路径下目录中的文件数

import os

def count_files(directory):
    count = 0
    for root, dirs, files in os.walk(directory):
        count += len(files)
    return count

directory = './relative/path/to/directory'
file_count = count_files(directory)
print(f"The number of files in the directory '{directory}' is: {file_count}")