Update find_weights.py

updated regex