Update test.py

test.py

@@ -89,8 +89,10 @@ def bert_encode(data):
 
 
 def test_predict(text):
+
+    text = clean_text(text)
     
-    test_encoded = bert_encode(text)
+    test_encoded = bert_encode([text])
     test_dataset = (
         tf.data.Dataset
         .from_tensor_slices((test_encoded))