Hemant0000 commited on
Commit
f68f46e
·
verified ·
1 Parent(s): 26db02c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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,