requirements updated

.gitignore

 *.jpg
 *.csv
 *.json
-*.png
+*.png
+*.zip

data/twitter/data_loader.py

 
-import cv2
-import pickle
 from PIL import Image
+from cv2 import cv2
 from numpy import asarray
 
 from data.data_loader import DatasetLoader

main.py

 import argparse
 
-from lime_main import lime_main
 from optuna_main import optuna_main
 from test_main import test_main
 from torch_main import torch_main
         optuna_main(config, args.use_optuna)
     elif args.just_test is not None:
         test_main(config, args.just_test)
-    elif args.use_lime is not None:
-        lime_main(config, args.use_lime)
     else:
         torch_main(config)
 

requirements.txt

 pytorch-lightning==1.2.10
 pytorch-model-summary==0.1.2
 pytorchtools==0.0.2
-scikit-image==0.17.2
 scikit-learn==0.24.2
 scipy==1.5.4
 seaborn==0.11.1
 torchsummary==1.5.1
 torchvision==0.9.1
 tqdm==4.60.0
-transformers==4.5.1
+tokenizers==0.13.3
+transformers==4.30.0
+timm==0.4.5
+albumentations==0.5.2