Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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('
|
|
|
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 |
|