Make a list of all jpg files in a folder

maketxt.py

import os
files = [f for f in os.listdir('.') if os.path.isfile(f)]
cwd = os.getcwd()
text_file = open("images.txt", "w")
print("Writing to text file")
for f in files:
    if "jpg" in f:
        text_file.write(cwd + "\\" + f + '\n')
        #text_file.close()
print("Done writing")