Update app.py
Browse files
app.py
CHANGED
@@ -3,6 +3,11 @@ from tensorflow.keras.applications import MobileNetV2
|
|
3 |
from tensorflow.keras import layers, models
|
4 |
import tensorflow as tf
|
5 |
|
|
|
|
|
|
|
|
|
|
|
6 |
# Image data generator with augmentation
|
7 |
train_datagen = ImageDataGenerator(
|
8 |
rescale=1.0/255,
|
|
|
3 |
from tensorflow.keras import layers, models
|
4 |
import tensorflow as tf
|
5 |
|
6 |
+
labels = ['Actinic Keratosis', 'Atopic Dermatitis', 'Benign Keratosis',
|
7 |
+
'Dermatofibroma', 'Melanocytic nevus', 'Melanoma',
|
8 |
+
'Squamous cell carcinoma', 'Tinea Ringworm candidiasis',
|
9 |
+
'Vascular lesion']
|
10 |
+
|
11 |
# Image data generator with augmentation
|
12 |
train_datagen = ImageDataGenerator(
|
13 |
rescale=1.0/255,
|