Hugging Face's logo

Pankaj001
/
ImageClassification-BrainTumor_data

Image Classification
Model card Files
xet
Community
Pankaj001 commited on
Commit
236d9e0
·
verified ·
1 Parent(s): 0fd6801

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -37,7 +37,7 @@ import tensorflow as tf
37
  import numpy as np
38
 
39
  # Load the pre-trained model
40
- model_name = "model/brain_tumor_model." # Replace with the actual model name
41
  model = tf.keras.models.load_model(model_name)
42
 
43
  # to get prediction
 
37
  import numpy as np
38
 
39
  # Load the pre-trained model
40
+ model_name = "model/brain_tumor_model.h5" # Replace with the actual model name
41
  model = tf.keras.models.load_model(model_name)
42
 
43
  # to get prediction