Spaces:
Sleeping
Sleeping
Commit
·
a35a773
1
Parent(s):
8a2aab0
Update model.py
Browse files
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("
|
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]
|