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:
|