NVASAIKUMAR commited on
Commit
de61903
·
1 Parent(s): 3de9dd3

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -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.h5",compile=False)
79
  result = model.predict(np.array([img]))
80
  ans = np.argmax(result)
81
  return lt[ans]
 
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]