Yadvendra commited on
Commit
0d7fabd
·
verified ·
1 Parent(s): 6465708

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from PIL import Image
6
  from sklearn.preprocessing import LabelEncoder
7
 
8
  # Load your pre-trained model (Make sure this matches the version used during training)
9
- model = tf.keras.models.load_model('dementia_cnn_model.h5')
10
 
11
  # Example class labels (update this list with your actual class labels)
12
  class_labels = ['glioma', 'pituitary', 'meningioma', 'healthy']
 
6
  from sklearn.preprocessing import LabelEncoder
7
 
8
  # Load your pre-trained model (Make sure this matches the version used during training)
9
+ model = tf.keras.models.load_model('brain_tumor_model.h5')
10
 
11
  # Example class labels (update this list with your actual class labels)
12
  class_labels = ['glioma', 'pituitary', 'meningioma', 'healthy']