v1.2

app.py

@@ -1,9 +1,6 @@
 import gradio as gr
 from main import main_fn
 
-#Input structure
-##Postal_,age_,town_,storey_,room_ = 680705, 30, 'CHOA CHU KANG', 12, '5 ROOM'
-
 
 iface = gr.Interface(
     fn=main_fn, 

main.py

@@ -5,7 +5,7 @@ import keras
 
 def main_fn(Name_):
 ##Input structure into model is##
-    filename = "training_2/gender_v1_freezebert.h5"
+    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')
@@ -16,10 +16,10 @@ def main_fn(Name_):
     female_prob = prob[0][0]
     if female_prob>0.5:
         gender = 'female'
-        prob = female_prob
+        prob = int(female_prob*100)
     else:
         gender = 'male'
-        prob = (1-female_prob)
+        prob = int((1-female_prob)*100)
     return gender, prob
     
 if __name__ == "__main__":