Remove un-used import on flask