Updating docstring, loading local model weights and adding parameters baa2ff5 Gflorent commited on Aug 17, 2022