Hugging Face's logo

Spaces:
kumar989
/
HealthVision
Sleeping

App Files Community
kumar989 commited on
Commit
5becdd7
·
1 Parent(s): 11da077

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +5 -5
model.py CHANGED
@@ -43,7 +43,7 @@ def bone_net(img):
43
  def brain_net(img):
44
  lt = ['pituitary', 'notumor', 'meningioma', 'glioma']
45
  img = cv2.resize(img,(224,224))
46
- model = tf.keras.models.load_model("Models/model_brain_tumur.h5")
47
  result = model.predict(np.array([img]))
48
  ans = np.argmax(result)
49
  return lt[ans]
@@ -51,7 +51,7 @@ def brain_net(img):
51
  def chest_net(img):
52
  lt = ['PNEUMONIA', 'NORMAL']
53
  img = cv2.resize(img,(224,224))
54
- model = tf.keras.models.load_model("Models/chest_cls_model.h5")
55
  result = model.predict(np.array([img]))
56
  ans = np.argmax(result)
57
  return lt[ans]
@@ -59,7 +59,7 @@ def chest_net(img):
59
  def Eye_net(img):
60
  lt = ['glaucoma', 'normal', 'diabetic_retinopathy', 'cataract']
61
  img = cv2.resize(img,(224,224))
62
- model = tf.keras.models.load_model("Models/Eye_diseases.h5")
63
  result = model.predict(np.array([img]))
64
  ans = np.argmax(result)
65
  return lt[ans]
@@ -67,7 +67,7 @@ def Eye_net(img):
67
  def kidney_net(img):
68
  lt = ['Cyst', 'Tumor', 'Stone', 'Normal']
69
  img = cv2.resize(img,(224,224))
70
- model = tf.keras.models.load_model("Models/kidney_stone.h5")
71
  result = model.predict(np.array([img]))
72
  ans = np.argmax(result)
73
  return lt[ans]
@@ -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("Models/skin_cancer.h5")
79
  result = model.predict(np.array([img]))
80
  ans = np.argmax(result)
81
  return lt[ans]
 
43
  def brain_net(img):
44
  lt = ['pituitary', 'notumor', 'meningioma', 'glioma']
45
  img = cv2.resize(img,(224,224))
46
+ model = tf.keras.models.load_model("Models/model_brain_tumur.h5",compile=False)
47
  result = model.predict(np.array([img]))
48
  ans = np.argmax(result)
49
  return lt[ans]
 
51
  def chest_net(img):
52
  lt = ['PNEUMONIA', 'NORMAL']
53
  img = cv2.resize(img,(224,224))
54
+ model = tf.keras.models.load_model("Models/chest_cls_model.h5",compile=False)
55
  result = model.predict(np.array([img]))
56
  ans = np.argmax(result)
57
  return lt[ans]
 
59
  def Eye_net(img):
60
  lt = ['glaucoma', 'normal', 'diabetic_retinopathy', 'cataract']
61
  img = cv2.resize(img,(224,224))
62
+ model = tf.keras.models.load_model("Models/Eye_diseases.h5",compile=False)
63
  result = model.predict(np.array([img]))
64
  ans = np.argmax(result)
65
  return lt[ans]
 
67
  def kidney_net(img):
68
  lt = ['Cyst', 'Tumor', 'Stone', 'Normal']
69
  img = cv2.resize(img,(224,224))
70
+ model = tf.keras.models.load_model("Models/kidney_stone.h5",compile=False)
71
  result = model.predict(np.array([img]))
72
  ans = np.argmax(result)
73
  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("Models/skin_cancer.h5",compile=False)
79
  result = model.predict(np.array([img]))
80
  ans = np.argmax(result)
81
  return lt[ans]