Update app.py

app.py

@@ -18,7 +18,7 @@ def predict_image(image):
     label = result[0]['label']
     confidence = result[0]['score']
 
-    prediction = 'female' if label == 'Label_0' else 'male'
+    prediction = 'female' if label == 'LABEL_0' else 'male'
     print(f"Prediction: {prediction}, Confidence: {confidence}")
 
     return label, confidence