Update app.py

app.py

@@ -20,12 +20,16 @@ def detect(img):
     img = np.expand_dims(img, axis=0)
     img = img/255
     prediction = model.predict(img)[0]
-    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)