Make demo.py's dataset optional.