Update gap_nmf.py