Fixing the app to use non-generic model building

Fixing the app to use non-generic model building