main.py

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_)