Update app.py

app.py

@@ -69,6 +69,14 @@ classes = {4: ('nv', ' melanocytic nevi'), 6: ('mel', 'melanoma'),
            0: ('akiec', 'Actinic keratoses and intraepithelial carcinomae'),
            3: ('df', 'dermatofibroma')}
 
+classes_map = {'nv': 'melanocytic nevi', 
+               'mel': 'melanoma',
+           'bkl', :'benign keratosis-like lesions',
+               'bcc':' basal cell carcinoma',
+           'vasc': ' pyogenic granulomas and hemorrhage',
+           'akiec': 'Actinic keratoses and intraepithelial carcinomae',
+           'df': 'dermatofibroma'}
+
 
 if file is not None:
 
@@ -116,7 +124,7 @@ if file is not None:
         if file:
             name = file.name.split(".")[0]
         if name in true_file['image_id'].values: 
-            st.write("True Label: ", true_file.loc[true_file['image_id']==name, 'dx'].iloc[0])
+            st.write("True Label: ", classes_map[true_file.loc[true_file['image_id']==name, 'dx'].iloc[0]])
         else:
             st.write("No match")