Update app.py

app.py

@@ -21,20 +21,17 @@ def detect(img):
     img = img/255
     prediction = model.predict(img)[0]
 
-    # Définition des seuils pour chaque classe
-    seuil_infection_bacterienne = 0.5
-    seuil_poumon_sain = 0.5
-    seuil_infection_biologique = 0.5
+
 
     # Initialisation du texte
     texte = ""
 
     # Détermination du texte et de la couleur pour chaque classe
-    if format_decimal(prediction[0]) >= seuil_infection_bacterienne:
+    if format_decimal(prediction[0]) >= 0.5:
         texte += '<span style="color:red;">Risque d\'infection bactérienne</span><br>'
-    if format_decimal(prediction[1]) >= seuil_poumon_sain:
+    if format_decimal(prediction[1]) >= 0.5:
         texte += '<span style="color:green;">Poumon sain</span><br>'
-    if format_decimal(prediction[2]) >= seuil_infection_biologique:
+    if format_decimal(prediction[2]) >= 0.5:
         texte += '<span style="color:orange;">Risque d\'infection biologique</span><br>'
     if texte == "":
         texte = "Classe indéterminée"