Update app.py

app.py

@@ -21,9 +21,9 @@ def predict(image):
     # return predicted_class
 
     probabilities = F.softmax(outputs, dim=1)
-    confidence_score = probabilities[0][predicted_class_idx].item()
     predicted_class_idx = probabilities.argmax(-1).item()
     predicted_class = class_names[predicted_class_idx]
+    confidence_score = probabilities[0][predicted_class_idx].item()
     
     return predicted_class, confidence_score