ybelkada
add model weights
038d8e2