Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ def detect(img):
|
|
17 |
# Obtenir la classe de la détection avec la probabilité la plus élevée
|
18 |
detection = detections[0] # On prend la première détection (peut être ajusté)
|
19 |
class_index = int(detection.cls[0]) # Obtenir l'indice de la classe
|
20 |
-
class_names = ['AMAZONE', 'BIOGUERRA', '
|
21 |
class_name = class_names[class_index] # Nom de la classe prédite
|
22 |
confidence = format(detection.conf[0], ".2f") # Confiance de la détection
|
23 |
return f"Classe : {class_name}, Confiance : {confidence}"
|
|
|
17 |
# Obtenir la classe de la détection avec la probabilité la plus élevée
|
18 |
detection = detections[0] # On prend la première détection (peut être ajusté)
|
19 |
class_index = int(detection.cls[0]) # Obtenir l'indice de la classe
|
20 |
+
class_names = ['AMAZONE', 'BIOGUERRA','PORTE DU NON RETOUR', 'REVENANT', 'ZANGBETO']
|
21 |
class_name = class_names[class_index] # Nom de la classe prédite
|
22 |
confidence = format(detection.conf[0], ".2f") # Confiance de la détection
|
23 |
return f"Classe : {class_name}, Confiance : {confidence}"
|