def main():
# os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
#
# with open(args.zipfile) as file:
# print(file)
#
# if args.zipfile[-4:] == '.zip':
# with zipfile.ZipFile(args.zipfile,"r") as zip_ref:
# zip_ref.extractall('extracted')
# elif args.zipfile[-3:] == '.7z':
# archive = py7zr.SevenZipFile(args.zipfile, mode='r')
# archive.extractall('extracted')
# archive.close()
# else:
# print("Cannot read the zip file")
dirName = 'extracted'
listOfFiles = getListOfFiles(dirName)
reader = Reader()
for elem in listOfFiles:
print(elem)
reader.predict(elem)
files = glob.glob('extracted/*')
print('---------------------NOW DELETE------------------------')
for elem in listOfFiles:
os.remove(elem)
for f in files:
print(f)
try:
shutil.rmtree(f)
except OSError as e:
print("Error: %s - %s." % (e.filename, e.strerror))
