Update app.py
Browse files
app.py
CHANGED
|
@@ -22,15 +22,53 @@ model.eval()
|
|
| 22 |
|
| 23 |
# Actual class names (replace generic labels)
|
| 24 |
class_names = [
|
| 25 |
-
"
|
| 26 |
-
"
|
| 27 |
-
"
|
| 28 |
-
"
|
| 29 |
-
"
|
| 30 |
-
"
|
| 31 |
-
"
|
| 32 |
-
"
|
| 33 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 34 |
]
|
| 35 |
|
| 36 |
# Transforms (must match training/validation)
|
|
|
|
| 22 |
|
| 23 |
# Actual class names (replace generic labels)
|
| 24 |
class_names = [
|
| 25 |
+
"Skin_Acne",
|
| 26 |
+
"Skin_Actinic Keratosis",
|
| 27 |
+
"Skin_Basal Cell Carcinoma",
|
| 28 |
+
"Skin_Chickenpox",
|
| 29 |
+
"Skin_Dermato Fibroma",
|
| 30 |
+
"Skin_Dyshidrotic Eczema",
|
| 31 |
+
"Skin_Melanoma",
|
| 32 |
+
"Skin_Nail Fungus",
|
| 33 |
+
"Skin_Nevus",
|
| 34 |
+
"Skin_Normal Skin",
|
| 35 |
+
"Skin_Pigmented Benign Keratosis",
|
| 36 |
+
"Skin_Ringworm",
|
| 37 |
+
"Skin_Seborrheic Keratosis",
|
| 38 |
+
"Skin_Squamous Cell Carcinoma",
|
| 39 |
+
"Skin_Vascular Lesion",
|
| 40 |
+
"all_benign",
|
| 41 |
+
"all_early",
|
| 42 |
+
"all_pre",
|
| 43 |
+
"all_pro",
|
| 44 |
+
"brain_glioma_tumor",
|
| 45 |
+
"brain_meningioma_tumor",
|
| 46 |
+
"brain_normal",
|
| 47 |
+
"brain_pituitary_tumor",
|
| 48 |
+
"breast_benign",
|
| 49 |
+
"breast_malignant",
|
| 50 |
+
"cervix_dyk",
|
| 51 |
+
"cervix_koc",
|
| 52 |
+
"cervix_mep",
|
| 53 |
+
"cervix_pab",
|
| 54 |
+
"cervix_sfi",
|
| 55 |
+
"colon_aca",
|
| 56 |
+
"colon_bnt",
|
| 57 |
+
"kidney_cyst",
|
| 58 |
+
"kidney_normal",
|
| 59 |
+
"kidney_stone",
|
| 60 |
+
"kidney_tumor",
|
| 61 |
+
"lung_colon_aca",
|
| 62 |
+
"lung_colon_n",
|
| 63 |
+
"lung_lung_aca",
|
| 64 |
+
"lung_lung_scc",
|
| 65 |
+
"lymph_cll",
|
| 66 |
+
"lymph_fl",
|
| 67 |
+
"lymph_mcl",
|
| 68 |
+
"oral_normal",
|
| 69 |
+
"oral_scc",
|
| 70 |
+
"pancreatic_normal",
|
| 71 |
+
"pancreatic_tumor"
|
| 72 |
]
|
| 73 |
|
| 74 |
# Transforms (must match training/validation)
|