Enhance GPU processing and batching capabilities; update README for clarity on test-time augmentation and batch sizes

First Push