Update app.py

app.py

@@ -25,7 +25,8 @@ if uploaded_file is not None:
     st.write("Classifying...")
     results = classifier(image)
 
-    # Display results
-    st.subheader("Results:")
-    for result in results:
-        st.write(f"**{result['label']}** - Score: {result['score']:.4f}")
+    # Get the label with the highest score
+    if results:
+        best_result = max(results, key=lambda x: x["score"])
+        st.subheader("Predicted Age Category:")
+        st.write(f"🟢 **{best_result['label']}** (Confidence: {best_result['score']:.4f})")