Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ def classify_image(image):
|
|
24 |
class_labels = ['Chihuahua', 'Muffin'] # Assumendo che tu abbia 2 classi
|
25 |
|
26 |
# Creare un dizionario con le percentuali di confidenza
|
27 |
-
confidence_scores = {class_labels[i]: float(prediction[i])
|
28 |
|
29 |
return confidence_scores # Restituire le percentuali per ciascuna classe
|
30 |
|
|
|
24 |
class_labels = ['Chihuahua', 'Muffin'] # Assumendo che tu abbia 2 classi
|
25 |
|
26 |
# Creare un dizionario con le percentuali di confidenza
|
27 |
+
confidence_scores = {class_labels[i]: float(prediction[i]) for i in range(len(class_labels))}
|
28 |
|
29 |
return confidence_scores # Restituire le percentuali per ciascuna classe
|
30 |
|