Spaces:
Sleeping
Sleeping
Commit
·
de61903
1
Parent(s):
3de9dd3
Update model.py
Browse files
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("
|
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]
|