Spaces:
Sleeping
Sleeping
Commit
·
95f692d
1
Parent(s):
7433624
Update model.py
Browse files
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,(
|
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)
|