Update app.py

app.py

@@ -148,8 +148,15 @@ for key in list(state_dict.keys()):
 model.load_state_dict(state_dict)
 model.eval()
 
-class_names = ['akk', 'bcc', 'bkl', 'df', 'mel','nv','vasc']
-class_names.sort()
+class_names = {
+    'akk': 'Actinic Keratosis',
+    'bcc': 'Basal Cell Carcinoma',
+    'bkl': 'Benign Keratosis',
+    'df': 'Dermatofibroma',
+    'mel': 'Melanoma',
+    'nv': 'Melanocytic Nevi',
+    'vasc': 'Vascular Lesion'
+}
 
 examples_dir = "sample"