Horus7 commited on
Commit
2679436
·
1 Parent(s): fb887f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -8
app.py CHANGED
@@ -8,18 +8,14 @@ from keras.models import load_model
8
  from tensorflow.keras.utils import load_img
9
 
10
  # Charger le modèle
11
- model = load_model('/content/drive/MyDrive/T-DEV-810/model_cv.h5')
 
12
 
13
- # Charger l'image
14
- img = load_img('/content/drive/MyDrive/T-DEV-810/DS/test/NORMAL/IM-0063-0001.jpeg', target_size=(100, 100))
15
- # Prétraiter l'image
16
- # img = image.img_to_array(img)
17
- img = np.expand_dims(img, axis=0)
18
- img = img/255
19
 
20
  def detect(img):
 
 
21
  prediction = model.predict(img)[0]
22
- print(prediction)
23
  if prediction[0] <= 0.80:
24
  return "Pneumonia Detected!"
25
 
 
8
  from tensorflow.keras.utils import load_img
9
 
10
  # Charger le modèle
11
+ model = load_model('model_cv.h5')
12
+
13
 
 
 
 
 
 
 
14
 
15
  def detect(img):
16
+ img = np.expand_dims(img, axis=0)
17
+ img = img/255
18
  prediction = model.predict(img)[0]
 
19
  if prediction[0] <= 0.80:
20
  return "Pneumonia Detected!"
21