Optimized the code for better model loading and faster inference for HF 83234c3 Tamal321 commited on Apr 20