Update app.py

app.py

@@ -8,18 +8,24 @@ from keras.models import load_model
 from tensorflow.keras.utils import load_img
 
 # Charger le modèle
-model = load_model('model_cv.h5')
+model = load_model('model_multi.h5')
 
 
 
 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!"
+    print(prediction)
+    
+    def format_decimal(value):
+        decimal_value = format(value, ".2f")
+        return decimal_value
+    
+    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)