AhmedBadrDev commited on
Commit
8dd048d
·
1 Parent(s): f1f7e22

update classes names

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -7,8 +7,9 @@ model = tf.keras.models.load_model('model.h5')
7
 
8
  # Define the class names
9
  class_names = {
10
- 0: 'Esophagitis',
11
- 1: 'Dyed lifted polyps'
 
12
  }
13
 
14
 
 
7
 
8
  # Define the class names
9
  class_names = {
10
+ 0: 'Glioma',
11
+ 1: 'Menin',
12
+ 2: 'Tumor'
13
  }
14
 
15