fixing import issue in text classification example

[parmeet]

Fixing import issue according to latest changes

[codecov]

Codecov Report

Merging #1256 (d027ab6) into master (50bd1b3) will not change coverage.
The diff coverage is n/a.

❗ Current head d027ab6 differs from pull request most recent head 257db3a. Consider uploading reports for the commit 257db3a to get more accurate results

@@           Coverage Diff           @@
##           master    #1256   +/-   ##
=======================================
  Coverage   78.80%   78.80%           
=======================================
  Files          67       67           
  Lines        3624     3624           
=======================================
  Hits         2856     2856           
  Misses        768      768           

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 50bd1b3...257db3a. Read the comment docs.