Update app.py

app.py

@@ -14,10 +14,15 @@ random.seed(0)
 each_image_name = os.listdir(image_path)
 random.shuffle(each_image_name)
 
-label_names = ['epiglottis', 'vocal cord', 'trachea', 'carina', 'right main bronchus', 'intermediate bronchus',
-               'right upper lobar bronchus', 'right middle lobar bronchus', 'right lower lobar bronchus', 'right superior segment bronchus',
-               'right basal bronchus', 'left main bronchus', 'left upper lobar bronchus', 'left division bronchus',
-               'left lingular bronchus', 'left lower bronchus', 'left superior segment', 'left basal bronchus']
+# label_names = ['epiglottis', 'vocal cord', 'trachea', 'carina', 'right main bronchus', 'intermediate bronchus',
+#                'right upper lobar bronchus', 'right middle lobar bronchus', 'right lower lobar bronchus', 'right superior segment bronchus',
+#                'right basal bronchus', 'left main bronchus', 'left upper lobar bronchus', 'left division bronchus',
+#                'left lingular bronchus', 'left lower bronchus', 'left superior segment', 'left basal bronchus']
+
+label_names = ['Epiglottis', 'Vocal Fold', 'Trachea', 'Left Main Bronchus', 'Carina', 'Right Main Bronchus', 'Left Upper Lobar Bronchus',
+               'Left Lower Bronchus', 'Right Upper Lobar Bronchus', 'Intermediate Bronchus', 'Right Lower Lobar Bronchus',
+               'Left Divsion Bronchus', 'Left Lingular Bronchus', 'Left Superior Segment',
+               'Left Basal Bronchus', 'Right Middle Lobar Bronchus', 'Right Basal Bronchus', 'Right Superior Segment Bronchus']
 
 model_list = ['YOLO-V8',
               'YOLO-V7',