Import BERT package

import.py

from sklearn.model_selection import train_test_split
import pandas as pd
import tensorflow as tf
import tensorflow_hub as hub
from datetime import datetime

# Install and import BERT's libraries
!pip install bert-tensorflow
import bert
from bert import run_classifier
from bert import optimization
from bert import tokenization