Fix BaseModelOutputWithPooling in predict.py and data.py; optimize predict.py model loading 17032f4 gcrn2318 commited on May 16
Fix classify_scene: handle BaseModelOutputWithPooling from newer transformers 4b84b46 gcrn2318 commited on May 16