Update app.py

app.py

@@ -18,7 +18,7 @@ from PIL import Image
 from IPython.display import display
 import io
 import pathlib
-print(tf.config.list_physical_devices('GPU'))
+
 RoboflowAPI = os.environ['RoboflowAPI']
 RoboflowCocoAPI = os.environ['RoboflowCocoAPI']
 
@@ -61,7 +61,9 @@ test_record_fname = dataset + '/test/tortoise.tfrecord'
 train_record_fname = dataset + '/train/tortoise.tfrecord'
 label_map_pbtxt_fname = dataset + '/train/tortoise_label_map.pbtxt'
 PATH_TO_LABELS = dataset + '/train/tortoise_label_map.pbtxt'
-print(os.listdir(dataset + '/train'))
+with open(PATH_TO_LABELS, 'r') as f:
+  contents = f.read()
+print(contents)
 category_index = label_map_util.create_category_index_from_labelmap(PATH_TO_LABELS, use_display_name=False)