dependencies = ['torch', 'tqdm', 'boto3', 'requests', 'regex']

from pytorch_transformers import (
    AutoTokenizer, AutoModel, AutoModelWithLMHead, AutoModelForSequenceClassification, AutoModelForQuestionAnswering
)