add labels

app.py

@@ -130,6 +130,20 @@ model = keras.models.load_model(MODEL_PATH)
 images = os.listdir(IMAGE_DIR)
 images = [f'{IMAGE_DIR}/{image}' for image in images]
 
+# Labels
+labels = {'fatemeh_hafari_27022022_181119right.png': 'classI',
+ 'majid_mohammadkhanlu_19022022_175338right.png': 'mesioangular',
+ 'zahra_khodabandeh_17052022_162804right.png': 'mesioangular',
+ '133028019739531250right.png': 'classII',
+ 'khadijeh_fard_aghababaei_03012022_120330right.png': 'B',
+ 'moharam_ali_khamseh_01112021_194329right.png': 'B',
+ 'parniya_mohamadinasab_08112021_184639right.png': 'B',
+ 'afsaneh_khalaji_24022022_183000right.png': 'B',
+ 'sogand_rahmani_13072022_113636right.png': 'B',
+ 'fatemeh_mirzaei_13072022_101734right.png': 'mesioangular',
+ 'fatemeh_hoseini_24012022_111330right.png': 'C',
+ 'padisar_ebrahimi_13122021_111512right.png': 'mesioangular'}
+
 # UserInterface
 st.header("3rdMolar Segmentation")
 st.subheader("Select Image:")
@@ -140,6 +154,7 @@ if image_file is not None:
     img , output1 = segment_molar(image_file)
     output2 = measure_molar(image_file)
     
+    st.subheader(f"Label : {labels[image_file]}")
     st.image(img[0],width=850)
     st.image(output1,width=850)
     st.image(output2,width=850)