Update app.py

app.py

@@ -8,24 +8,19 @@ from keras.models import load_model
 from tensorflow.keras.utils import load_img
 
 # Charger le modèle
-model = load_model('model_multi.h5')
+
+model = load_model('model_cv.h5')
 
 
 
 def detect(img):
+    img = np.expand_dims(img, axis=0)
+    img = img/255
     prediction = model.predict(img)[0]
-    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é"
+    if prediction[0] <= 0.80:
+        return "Pneumonia  Detected!"
+
+    return "Pneumonia Not Detected!"
 
 
 # result = detect(img)