Spaces:
Sleeping
Sleeping
Update model.py
Browse files
model.py
CHANGED
@@ -7,7 +7,7 @@ def classify(img):
|
|
7 |
im = img
|
8 |
lt = ["other","Bone","Brain","eye","kidney","chest","skin"]
|
9 |
im = cv2.resize(im,(52,52))
|
10 |
-
model = tf.keras.models.load_model("Models/all-in-one.h5")
|
11 |
result = model.predict(np.array([im]))
|
12 |
a = np.argmax(result)
|
13 |
c=""
|
@@ -31,7 +31,7 @@ def classify(img):
|
|
31 |
|
32 |
def bone_net(img):
|
33 |
img = cv2.resize(img,(224,224))
|
34 |
-
model = tf.keras.models.load_model("Models/Fracture_detection.h5")
|
35 |
result = model.predict(np.array([img]))
|
36 |
op=""
|
37 |
if result[0]<0.5:
|
|
|
7 |
im = img
|
8 |
lt = ["other","Bone","Brain","eye","kidney","chest","skin"]
|
9 |
im = cv2.resize(im,(52,52))
|
10 |
+
model = tf.keras.models.load_model("Models/all-in-one.h5",compile=False)
|
11 |
result = model.predict(np.array([im]))
|
12 |
a = np.argmax(result)
|
13 |
c=""
|
|
|
31 |
|
32 |
def bone_net(img):
|
33 |
img = cv2.resize(img,(224,224))
|
34 |
+
model = tf.keras.models.load_model("Models/Fracture_detection.h5",compile=False)
|
35 |
result = model.predict(np.array([img]))
|
36 |
op=""
|
37 |
if result[0]<0.5:
|