udayjawheri
commited on
Commit
•
905e44e
1
Parent(s):
4162286
Upload 12 files
Browse files- .gitattributes +1 -0
- 00000.png +0 -0
- 00001.png +3 -0
- 00002.png +0 -0
- 00003.png +0 -0
- 00004.png +0 -0
- __pycache__/gradio.cpython-311.pyc +0 -0
- __pycache__/model.cpython-311.pyc +0 -0
- __pycache__/model.cpython-39.pyc +0 -0
- __pycache__/streamlit.cpython-311.pyc +0 -0
- app.py +54 -0
- model_age.h5 +3 -0
- model_gender.h5 +3 -0
.gitattributes
CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
+
00001.png filter=lfs diff=lfs merge=lfs -text
|
00000.png
ADDED
00001.png
ADDED
Git LFS Details
|
00002.png
ADDED
00003.png
ADDED
00004.png
ADDED
__pycache__/gradio.cpython-311.pyc
ADDED
Binary file (2.62 kB). View file
|
|
__pycache__/model.cpython-311.pyc
ADDED
Binary file (2.14 kB). View file
|
|
__pycache__/model.cpython-39.pyc
ADDED
Binary file (1.2 kB). View file
|
|
__pycache__/streamlit.cpython-311.pyc
ADDED
Binary file (395 Bytes). View file
|
|
app.py
ADDED
@@ -0,0 +1,54 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
import tensorflow as tf
|
3 |
+
from PIL import Image
|
4 |
+
import numpy as np
|
5 |
+
import os
|
6 |
+
import pandas as pd
|
7 |
+
|
8 |
+
model_gender = tf.keras.models.load_model('model_gender.h5')
|
9 |
+
model_age = tf.keras.models.load_model('model_age.h5')
|
10 |
+
|
11 |
+
actual_data = {
|
12 |
+
"000000.png": {"img": 1,"age": 85.0, "gender": "female"},
|
13 |
+
"000001.png": {"img": 2,"age": 72.0, "gender": "female"},
|
14 |
+
"000002.png": {"img": 3,"age": 45.0, "gender": "male"},
|
15 |
+
"000003.png": {"img": 4,"age": 59.0, "gender": "male"},
|
16 |
+
"000004.png": {"img": 5,"age": 37.0, "gender": "male"}
|
17 |
+
}
|
18 |
+
|
19 |
+
df = pd.DataFrame(actual_data).T
|
20 |
+
|
21 |
+
def preprocess_image(image):
|
22 |
+
# Assuming image is a PIL Image object from Gradio
|
23 |
+
img = image.convert('L') # Convert to grayscale
|
24 |
+
img = img.resize((128, 128))
|
25 |
+
img = np.array(img) / 255.0 # Normalize pixel values
|
26 |
+
img = img.reshape((1, 128, 128, 1)) # Add channel dimension
|
27 |
+
return img
|
28 |
+
|
29 |
+
def predict(image):
|
30 |
+
preprocessed_image = preprocess_image(image)
|
31 |
+
gender_pred = model_gender.predict(preprocessed_image)[0][0]
|
32 |
+
age_pred = model_age.predict(preprocessed_image)[0][0]
|
33 |
+
gender = "Male" if gender_pred > 0.5 else "Female"
|
34 |
+
list = "{:.2f}".format(age_pred),gender,df
|
35 |
+
return list
|
36 |
+
|
37 |
+
|
38 |
+
|
39 |
+
# Gradio Interface with separate outputs
|
40 |
+
text_age = gr.components.Textbox(label="Predicted Age")
|
41 |
+
text_gender = gr.components.Textbox(label="Predicted Gender")
|
42 |
+
|
43 |
+
interface = gr.Interface(predict, gr.components.Image(height=440,width=1000,label="Upload Image", type="pil"),
|
44 |
+
outputs=[text_age, text_gender, gr.DataFrame(value=df)],
|
45 |
+
examples=[
|
46 |
+
os.path.join(os.path.dirname(__file__),"00000.png"),
|
47 |
+
os.path.join(os.path.dirname(__file__),"00001.png"),
|
48 |
+
os.path.join(os.path.dirname(__file__),"00002.png"),
|
49 |
+
os.path.join(os.path.dirname(__file__),"00003.png"),
|
50 |
+
os.path.join(os.path.dirname(__file__),"00004.png")],
|
51 |
+
|
52 |
+
allow_flagging='never',
|
53 |
+
theme=gr.themes.Soft(),
|
54 |
+
title="Age and Gender Prediction").launch()
|
model_age.h5
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8cf2aeb631a124e81b75b704ad08c928aabb28bbbb1064859b3207bbeac6b965
|
3 |
+
size 2556856
|
model_gender.h5
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ebb54b7596daccca78a8e757c84b76a28b40b5d7da347e617850956877432a53
|
3 |
+
size 2439608
|