cybernatedArt commited on
Commit
462128d
1 Parent(s): 7ba9fc9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -13
app.py CHANGED
@@ -5,19 +5,19 @@ path_to_model = "./model_2.h5"
5
 
6
  model = tf.keras.models.load_model(path_to_model)
7
 
8
- labels = ['Acne and Rosacea Photos',
9
- 'Actinic Keratosis Basal Cell Carcinoma and other Malignant Lesions',
10
- 'Atopic Dermatitis Photos', 'Bullous Disease Photos',
11
- 'Cellulitis Impetigo and other Bacterial Infections',
12
- 'Eczema Photos', 'Exanthems and Drug Eruptions', 'Hair Loss Photos Alopecia and other Hair Diseases',
13
- 'Herpes HPV and other STDs Photos', 'Light Diseases and Disorders of Pigmentation',
14
- 'Lupus and other Connective Tissue diseases',
15
- 'Melanoma Skin Cancer Nevi and Moles', 'Nail Fungus and other Nail Disease',
16
- 'Poison Ivy Photos and other Contact Dermatitis',
17
- 'Psoriasis pictures Lichen Planus and related diseases', 'Scabies Lyme Disease and other Infestations and Bites',
18
- 'Seborrheic Keratoses and other Benign Tumors', 'Systemic Disease',
19
- 'Tinea Ringworm Candidiasis and other Fungal Infections',
20
- 'Urticaria Hives', 'Vascular Tumors', 'Vasculitis Photos', 'Warts Molluscum and other Viral Infections']
21
 
22
  def classify_image(photos):
23
  photos = photos.reshape((-1, 256, 256, 3))
 
5
 
6
  model = tf.keras.models.load_model(path_to_model)
7
 
8
+ labels = ['Acne and Rosacea',
9
+ 'Actinic Keratosis Basal Cell Carcinoma',
10
+ 'Atopic Dermatitis', 'Bullous Disease',
11
+ 'Cellulitis Impetigo (Bacterial Infections)',
12
+ 'Eczema', 'Exanthems and Drug Eruptions', 'Hair Loss (Alopecia)',
13
+ 'Herpes HPV', 'Disorders of Pigmentation',
14
+ 'Lupus ',
15
+ 'Melanoma (Skin Cancer)', 'Nail Fungus',
16
+ 'Poison Ivy',
17
+ 'Psoriasis (Lichen Planus)', 'Scabies Lyme',
18
+ 'Seborrheic Keratoses', 'Systemic Disease',
19
+ 'Tinea Ringworm (Fungal Infections)',
20
+ 'Urticaria Hives', 'Vascular Tumors', 'Vasculitis', 'Warts Molluscum']
21
 
22
  def classify_image(photos):
23
  photos = photos.reshape((-1, 256, 256, 3))