itsTomLie commited on
Commit
b7c1842
·
verified ·
1 Parent(s): 4bc0e7b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def predict_image(image):
18
  label = result[0]['label']
19
  confidence = result[0]['score']
20
 
21
- print(f"Prediction: {label}, Confidence: {confidence}")
22
 
23
  return label, confidence
24
 
 
18
  label = result[0]['label']
19
  confidence = result[0]['score']
20
 
21
+ print(f"Prediction: {"female" if label == "Label_0" else "male" }, Confidence: {confidence}")
22
 
23
  return label, confidence
24