Update consumer.py

Just added a switch into the consumer so that no uncommenting is necessary for using kafka 0.8.1