udayjawheri commited on
Commit
905e44e
1 Parent(s): 4162286

Upload 12 files

Browse files
.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

  • SHA256: cbc88d4519c80a044213fd01d1e7c08b03e215c63dae201df17791e384c8683b
  • Pointer size: 132 Bytes
  • Size of remote file: 1 MB
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