Upload fun_advaitbert.py

fun_advaitbert.py

@@ -213,10 +213,15 @@ 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=''
-                    pred_duty=''
                     pass
 
                 Var_CTH.append(pred_CTH)
@@ -257,7 +262,7 @@ def predict_CTH(txt):
                 f'9. {P9} {Q9} '
                 f'10. {P10} {Q10}')
 
-            print('output_str_msg',output_str_msg)
+            #print('output_str_msg',output_str_msg)
   
             return {str(P1):float(val_1),str(Q1):float(val_1),
                     str(P2):float(val_2),str(Q2):float(val_2),
@@ -279,7 +284,7 @@ def llm_model_function(txt,history,chatbot=[], temperature=0.9, max_new_tokens=1
 
     global output_str_msg
 
-    print('output_str_msg',output_str_msg)
+    #print('output_str_msg',output_str_msg)
 
     if output_str_msg!='Not a adequate description':