Update fun_advaitbert.py

fun_advaitbert.py

@@ -12,6 +12,7 @@ tf.get_logger().setLevel('ERROR')
 from huggingface_hub import InferenceClient
 import math
 import gradio as gr
+from datetime import datetime
 
 num_warmup_steps=1
 num_train_steps=1
@@ -163,6 +164,8 @@ additional_inputs=[
 
 def predict_CTH(txt):
     print('Desc: ',txt)
+    now = datetime.now()
+    print("Time =", now)
     if (txt!='') and len(txt)>=3 and (count_special_character(txt)):
         valid_data = tf.data.Dataset.from_tensor_slices(([txt] , [1])) # 1 refers to 'entertainment' and 2 refers to 'sport'
         valid_data = (valid_data.map(to_feature_map).batch(1))
@@ -201,8 +204,14 @@ def predict_CTH(txt):
 
                 try:
                     pred_duty=df_report_DUTY[df_report_DUTY['CTH']==str(pred_CTH)]['DUTY_RATE'].iloc[0]
+                except:
+                    pred_duty=''
+                    pass
+
+                try:
                     pred_desc=df_report[df_report['CTH Code']==str(pred_CTH)]['Concat Description'].iloc[0]
                 except:
+                    pred_desc=''
                     pass
 
                 Var_CTH.append(pred_CTH)