Update app.py

app.py

@@ -59,9 +59,8 @@ TEST_IMAGE_PATHS = sorted(list(PATH_TO_TEST_IMAGES_DIR.glob("*.jpg")))
 dataset = 'Tortoise'
 test_record_fname = dataset + '/test/tortoise.tfrecord'
 train_record_fname = dataset + '/train/tortoise.tfrecord'
-label_map_pbtxt_fname = 'tortoise_label_map.pbtxt'
-PATH_TO_LABELS = 'tortoise_label_map.pbtxt'
-print(os.listsir())
+label_map_pbtxt_fname = dataset + '/train/tortoise_label_map.pbtxt'
+PATH_TO_LABELS = dataset + '/train/tortoise_label_map.pbtxt'
 category_index = label_map_util.create_category_index_from_labelmap(PATH_TO_LABELS, use_display_name=False)