Revert "Update the Model in JS version"

This reverts commit 16d3a1d1b67b51b9fb2cada32f2b0ab96077418f.