Spaces:
Running
Running
File size: 898 Bytes
aab7ecd 0220add c06d25f 77b189e d5050b9 aab7ecd d5050b9 aab7ecd 6dec054 d5050b9 aab7ecd d5050b9 6dec054 d5050b9 6dec054 0220add aab7ecd d5050b9 0220add |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import os
import pandas as pd
# import keras
import tensorflow as tf
import tensorflow_hub as hub
import tensorflow_text
### This is to predict gender
def main_fn(Name_):
##Input structure into model is##
filename = "gender_v1_freezebert.h5"
if os.path.exists(filename):
model = tf.keras.models.load_model((filename), custom_objects={'KerasLayer':hub.KerasLayer})
print('loaded model')
else:
print('failed loading model')
prob = model.predict([Name_])
female_prob = prob[0][0]
if female_prob>0.5:
gender = 'female'
prob = int(female_prob*100)
else:
gender = 'male'
prob = int((1-female_prob)*100)
final_text = f"{gender}_confidence at {prob}%"
return final_text
#To start returning the probability confidence at later stage
if __name__ == "__main__":
Name_ = 'John'
final_text = main_fn(Name_)
|