Horus7 commited on
Commit
5abd417
1 Parent(s): 92a2212

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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', 'REVENANT', 'ZANGBETO', 'PORTE DU NON RETOUR']
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}"