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