hgrif commited on
Commit
5f80ca4
β€’
1 Parent(s): 5ddb621

Fix import

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -8,6 +8,7 @@ import streamlit as st
8
  from transformers import BertTokenizer, TFAutoModelForMaskedLM
9
  from rhyme_with_ai.utils import color_new_words, pairwise, sanitize
10
  from rhyme_with_ai.token_weighter import TokenWeighter
 
11
 
12
 
13
  DEFAULT_QUERY = "Machines will take over the world soon"
 
8
  from transformers import BertTokenizer, TFAutoModelForMaskedLM
9
  from rhyme_with_ai.utils import color_new_words, pairwise, sanitize
10
  from rhyme_with_ai.token_weighter import TokenWeighter
11
+ from rhyme_with_ai.rhyme import query_rhyme_words
12
 
13
 
14
  DEFAULT_QUERY = "Machines will take over the world soon"