Update app.py

app.py

@@ -12,15 +12,20 @@ from tensorflow.keras.utils import load_img
 model = load_model('model_cv.h5')
 
 
-
+def format_decimal(value):
+    decimal_value = format(value, ".2f")
+    return decimal_value
+    
 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 "Bactérie détectée"
+    if format_decimal(prediction[1]) >= "0.5":
+        return "Poumon sain"
+    if format_decimal(prediction[2]) >= "0.5":
+        return "Virus détecté"
 
 
 # result = detect(img)