Updated README with model details