Spaces:
Sleeping
Sleeping
Jyantkumar
commited on
Commit
•
74680d3
1
Parent(s):
f48de38
Update app.py
Browse files
app.py
CHANGED
@@ -48,7 +48,7 @@ def sepia(input_img_path):
|
|
48 |
img = img_to_array(img)
|
49 |
img = img / 255
|
50 |
img = img.reshape(1,224,224,3)
|
51 |
-
p = (
|
52 |
if p==0:
|
53 |
return "Men"
|
54 |
else:
|
|
|
48 |
img = img_to_array(img)
|
49 |
img = img / 255
|
50 |
img = img.reshape(1,224,224,3)
|
51 |
+
p = (model.predict(img)>=0.5).astype(int)[0][0]
|
52 |
if p==0:
|
53 |
return "Men"
|
54 |
else:
|