Fix permission issues by using a non-root user and dedicated cache directories c6e1c92 SaiMupparaju commited on Mar 11
Fix model ID to use 'gpt2' instead of 'gpt2-small' and add error handling f7aa3bd SaiMupparaju commited on Mar 11