Update demo.py

deleted import ai