NVASAIKUMAR commited on
Commit
95f692d
·
1 Parent(s): 7433624

Update model.py

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