Update compAnIonv1.py

compAnIonv1.py

@@ -29,7 +29,8 @@ import pandas as pd
 
 # Import Tensorflow and BERT models
 import tensorflow as tf
-from tensorflow import keras
+from tensorflow.keras 
+#from tensorflow import keras
 from tensorflow.keras import layers
 from transformers import BertTokenizer
 from transformers import TFBertModel
@@ -135,8 +136,8 @@ def run_inference_model(conversations):
 
         # Apply Model Prediction to testData
         y_pred = inference_model.predict(bert_inputs)
-        prediction = f_one_or_zero(y_pred)
-        return prediction
+        #prediction = f_one_or_zero(y_pred)
+        return y_pred