Update app.py

app.py

@@ -28,23 +28,19 @@ def detect(img):
     # Faire une prédiction
     prediction = model.predict(img)[0]
 
-    # Initialisation du texte
-    texte = ""
-
+    
     # Classes prédictives avec leurs index respectifs
     class_names = ['AMAZONE', 'BIOGUERRA', 'REVENANT', 'ZANGBETO']
 
-    # Détermination du texte pour chaque classe
-    for idx, class_name in enumerate(class_names):
-        if prediction[idx] >= 0.5:
-            texte += f'{class_name} détecté\n'
+    # Trouver l'indice de la classe avec la probabilité la plus élevée
+    class_index = np.argmax(prediction)
 
-    # Si aucune classe n'est détectée
-    if texte == "":
-        texte = "Classe indéterminée"
+    # Obtenir le nom de la classe prédite
+    texte = class_names[class_index]
 
     return texte
 
+
 title = "Orisha"
 
 iface = gr.Interface(