NVASAIKUMAR commited on
Commit
a35a773
·
1 Parent(s): 8a2aab0

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -43,7 +43,7 @@ def bone_net(img):
43
  def brain_net(img):
44
  lt = ['pituitary', 'notumor', 'meningioma', 'glioma']
45
  img = cv2.resize(img,(224,224))
46
- model = tf.keras.models.load_model("model_brain_tumur.h5",compile=False)
47
  result = model.predict(np.array([img]))
48
  ans = np.argmax(result)
49
  return lt[ans]
 
43
  def brain_net(img):
44
  lt = ['pituitary', 'notumor', 'meningioma', 'glioma']
45
  img = cv2.resize(img,(224,224))
46
+ model = tf.keras.models.load_model("brain.h5",compile=False)
47
  result = model.predict(np.array([img]))
48
  ans = np.argmax(result)
49
  return lt[ans]