Update app.py

app.py

@@ -27,11 +27,11 @@ def detect(img):
     texte = ""
 
     # Détermination du texte et de la couleur pour chaque classe
-    if format_decimal(prediction[0]) >= 0.5:
+    if format_decimal(prediction[0]) >= "0.5":
         texte += '<span style="color:red;">Risque d\'infection bactérienne</span><br>'
-    if format_decimal(prediction[1]) >= 0.5:
+    if format_decimal(prediction[1]) >= "0.5":
         texte += '<span style="color:green;">Poumon sain</span><br>'
-    if format_decimal(prediction[2]) >= 0.5:
+    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"