NVASAIKUMAR commited on
Commit
bdc7c4a
·
1 Parent(s): 30cf921

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -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("Fracture.h5",compile=False)
35
  result = model.predict(np.array([img]))
36
  op=""
37
  if result[0]<0.5:
@@ -75,7 +75,7 @@ def kidney_net(img):
75
  def skin_net(img):
76
  lt = ['pigmented benign keratosis', 'melanoma', 'vascular lesion', 'actinic keratosis', 'squamous cell carcinoma', 'basal cell carcinoma', 'seborrheic keratosis', 'dermatofibroma', 'nevus']
77
  # img = cv2.resize(img,(224,224))
78
- model = tf.keras.models.load_model("skin_cancer.h5",compile=False)
79
  result = model.predict(np.array([img]))
80
  ans = np.argmax(result)
81
  return lt[ans]
 
31
 
32
  def bone_net(img):
33
  # img = cv2.resize(img,(224,224))
34
+ model = tf.keras.models.load_model("fracture.h5",compile=False)
35
  result = model.predict(np.array([img]))
36
  op=""
37
  if result[0]<0.5:
 
75
  def skin_net(img):
76
  lt = ['pigmented benign keratosis', 'melanoma', 'vascular lesion', 'actinic keratosis', 'squamous cell carcinoma', 'basal cell carcinoma', 'seborrheic keratosis', 'dermatofibroma', 'nevus']
77
  # img = cv2.resize(img,(224,224))
78
+ model = tf.keras.models.load_model("skin.h5",compile=False)
79
  result = model.predict(np.array([img]))
80
  ans = np.argmax(result)
81
  return lt[ans]