kumar989 commited on
Commit
6beea6a
·
1 Parent(s): 5becdd7

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -2
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: