simran12m commited on
Commit
08819d8
1 Parent(s): 3f855ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,7 +25,7 @@ def getAgeGender(image_path):
25
  image = image.reshape((image.shape[0], image.shape[1], 1))
26
 
27
  # Loading the trained model:
28
- model = load_model('data.h5')
29
 
30
  # Getting the predictions:
31
  image = image / 255
@@ -35,7 +35,7 @@ def getAgeGender(image_path):
35
  return age, gender
36
 
37
  def main():
38
- st.title("Age and Gender Prediction with Streamlit")
39
 
40
  uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
41
 
 
25
  image = image.reshape((image.shape[0], image.shape[1], 1))
26
 
27
  # Loading the trained model:
28
+ model = load_model('data_old.h5')
29
 
30
  # Getting the predictions:
31
  image = image / 255
 
35
  return age, gender
36
 
37
  def main():
38
+ st.title("Age and Gender Prediction")
39
 
40
  uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
41