datasets
PyPDF2
torch