Spaces:
Build error
Build error
Fix import
Browse files
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"
|