Fix import

app.py

@@ -8,6 +8,7 @@ import streamlit as st
 from transformers import BertTokenizer, TFAutoModelForMaskedLM
 from rhyme_with_ai.utils import color_new_words, pairwise, sanitize
 from rhyme_with_ai.token_weighter import TokenWeighter
+from rhyme_with_ai.rhyme import query_rhyme_words
 
 
 DEFAULT_QUERY = "Machines will take over the world soon"