Update app.py
Browse files
app.py
CHANGED
|
@@ -55,10 +55,10 @@ def predict_gender(img):
|
|
| 55 |
|
| 56 |
preds = model.predict(x)
|
| 57 |
|
| 58 |
-
if preds[0]
|
| 59 |
-
return "Predicted female " + str(preds[0]) + "%"
|
| 60 |
-
else:
|
| 61 |
return "Predicted male " + str(preds[0]) + "%"
|
|
|
|
|
|
|
| 62 |
|
| 63 |
# Create an interface with a file input and a text output
|
| 64 |
interface = gr.Interface(
|
|
|
|
| 55 |
|
| 56 |
preds = model.predict(x)
|
| 57 |
|
| 58 |
+
if preds[0] > 0.1:
|
|
|
|
|
|
|
| 59 |
return "Predicted male " + str(preds[0]) + "%"
|
| 60 |
+
else:
|
| 61 |
+
return "Predicted female " + str(preds[0]) + "%"
|
| 62 |
|
| 63 |
# Create an interface with a file input and a text output
|
| 64 |
interface = gr.Interface(
|