Update app.py

app.py

@@ -26,25 +26,20 @@ def detect(img):
     seuil_poumon_sain = 0.5
     seuil_infection_biologique = 0.5
 
-    # Initialisation du texte et de la couleur
+    # Initialisation du texte
     texte = ""
-    couleur = ""
 
     # Détermination du texte et de la couleur pour chaque classe
     if format_decimal(prediction[0]) >= seuil_infection_bacterienne:
-        texte = "Risque d'infection bactérienne"
-        couleur = "red"
-    elif format_decimal(prediction[1]) >= seuil_poumon_sain:
-        texte = "Poumon sain"
-        couleur = "green"
-    elif format_decimal(prediction[2]) >= seuil_infection_biologique:
-        texte = "Risque d'infection biologique"
-        couleur = "orange"
-    else:
+        texte += '<span style="color:red;">Risque d\'infection bactérienne</span><br>'
+    if format_decimal(prediction[1]) >= seuil_poumon_sain:
+        texte += '<span style="color:green;">Poumon sain</span><br>'
+    if format_decimal(prediction[2]) >= seuil_infection_biologique:
+        texte += '<span style="color:orange;">Risque d\'infection biologique</span><br>'
+    if texte == "":
         texte = "Classe indéterminée"
-        couleur = "black"
 
-    return texte, couleur
+    return texte
 
 
 
@@ -60,5 +55,5 @@ input = gr.inputs.Image(shape=(100,100))
 
 title = "PneumoDetect: Detection de pneumonie par x-ray"
 
-iface = gr.Interface(fn=detect, inputs=input, outputs=["text", "color"],examples = examples, examples_per_page=20, title=title)
+iface = gr.Interface(fn=detect, inputs=input, outputs="text",examples = examples, examples_per_page=20, title=title)
 iface.launch(inline=False)