Spaces:
Sleeping
Sleeping
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_) | |