Update main.py

commented line having code using sklearn