diff --git a/print_labels_from_csv.py b/print_labels_from_csv.py index 51c7e60..33753a0 100644 --- a/print_labels_from_csv.py +++ b/print_labels_from_csv.py @@ -6,7 +6,7 @@ import sys medir = os.path.dirname(__file__) -if len(sys.argv) > 2: +if len(sys.argv) > 1: example_data_path = sys.argv[1] else: example_data_path = os.path.join(medir, 'labels.csv') @@ -23,4 +23,4 @@ for _, row in df.iterrows(): if row['Cut'] == 1: cut = True printer.print_image(file_path, cut = cut, rotation=270) - os.remove(file_path) \ No newline at end of file + os.remove(file_path)