Update app.py

app.py

@@ -9,7 +9,7 @@ from tensorflow.keras.utils import load_img
 
 # Charger le modèle
 
-model = load_model('model_cv.h5')
+model = load_model('model_multi.h5')
 
 
 def format_decimal(value):
@@ -20,16 +20,16 @@ def detect(img):
     img = np.expand_dims(img, axis=0)
     img = img/255
     prediction = model.predict(img)[0]
-        if prediction[0] <= 0.80:
-        return "Pneumonia  Detected!"
-
-    return "Pneumonia Not Detected!"
-    # if format_decimal(prediction[0]) >= "0.5":
-    #     return "Risque d'infection bactérienne"
-    # if format_decimal(prediction[1]) >= "0.5":
-    #     return "Poumon sain"
-    # if format_decimal(prediction[2]) >= "0.5":
-    #     return "Risque d'infection biologique"
+    #     if prediction[0] <= 0.80:
+    #     return "Pneumonia  Detected!"
+
+    # return "Pneumonia Not Detected!"
+    if format_decimal(prediction[0]) >= "0.5":
+        return "Risque d'infection bactérienne"
+    if format_decimal(prediction[1]) >= "0.5":
+        return "Poumon sain"
+    if format_decimal(prediction[2]) >= "0.5":
+        return "Risque d'infection biologique"
 
 
 # result = detect(img)