KhadijaAsehnoune12 commited on
Commit
da42d5c
·
verified ·
1 Parent(s): dfdf045

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,10 +40,10 @@ def predict(image):
40
 
41
  # Obtenir le label et le score de confiance de la classe la plus probable
42
  predicted_label = id2label[str(predicted_class_idx)]
43
- confidence_score = probs[predicted_class_idx].item()
44
 
45
  # Retourner le label et le score de confiance
46
- return f"{predicted_label}: {confidence_score:.2f}"
47
 
48
  # Créer l'interface Gradio
49
  image = gr.Image(type="pil")
 
40
 
41
  # Obtenir le label et le score de confiance de la classe la plus probable
42
  predicted_label = id2label[str(predicted_class_idx)]
43
+ confidence_score = probs[predicted_class_idx].item() * 100 # Multiplie par 100 pour obtenir un pourcentage
44
 
45
  # Retourner le label et le score de confiance
46
+ return f"{predicted_label}: {confidence_score:.2f}%"
47
 
48
  # Créer l'interface Gradio
49
  image = gr.Image(type="pil")