Hugging Face's logo

Spaces:
Dineshkumars
/
Brain-Tumor-Classification
Runtime error

App Files Community
Dineshkumars commited on
Commit
ce0f610
•
1 Parent(s): dac2b09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import tensorflow
2
  from tensorflow import keras
3
  import gradio as gr
4
 
5
- model = keras.models.load_model('simple_model.h5')
6
  potato_classes = ['Glioma','Meningioma','No Tumor','Pituitary']
7
 
8
  def predict_input_image(img):
 
2
  from tensorflow import keras
3
  import gradio as gr
4
 
5
+ model = keras.models.load_model('brain_vgg.h5')
6
  potato_classes = ['Glioma','Meningioma','No Tumor','Pituitary']
7
 
8
  def predict_input_image(img):