Update app.py

app.py

@@ -21,7 +21,7 @@ def predict_image(image):
     prediction = 'female' if label == 'LABEL_0' else 'male'
     print(f"Prediction: {prediction}, Confidence: {confidence}")
 
-    return label, confidence
+    return prediction, confidence
 
 example_images = [
     os.path.join("examples", img_name) for img_name in sorted(os.listdir("examples"))