KhadijaAsehnoune12
commited on
Update app.py
Browse files
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")
|