Spaces:
Build error
Build error
Revert "Switch to model that's not finetuned"
Browse filesThis reverts commit aab7a6efeb804c6f2ad019a0068a9b00cca11706.
app.py
CHANGED
|
@@ -21,7 +21,7 @@ ITER_FACTOR = 5
|
|
| 21 |
|
| 22 |
LANGUAGE = st.sidebar.radio("Language", ["english", "dutch"],0)
|
| 23 |
if LANGUAGE == "english":
|
| 24 |
-
MODEL_PATH = "bert-large-
|
| 25 |
elif LANGUAGE == "dutch":
|
| 26 |
MODEL_PATH = "GroNLP/bert-base-dutch-cased"
|
| 27 |
else:
|
|
@@ -305,7 +305,6 @@ def query_rhyme_words(sentence: str, n_rhymes: int, language:str="english") -> L
|
|
| 305 |
if language == "english":
|
| 306 |
return query_datamuse_api(last_word, n_rhymes)
|
| 307 |
elif language == "dutch":
|
| 308 |
-
print('hai')
|
| 309 |
return mick_rijmwoordenboek(last_word, n_rhymes)
|
| 310 |
else:
|
| 311 |
raise NotImplementedError(f"Unsupported language ({language}) expected 'english' or 'dutch'.")
|
|
@@ -338,14 +337,13 @@ def mick_rijmwoordenboek(word: str, n_words: int):
|
|
| 338 |
html = get(url)
|
| 339 |
soup = Soup(html)
|
| 340 |
|
| 341 |
-
results = soup.find("div", {"id": "rhymeResultsWords"}).html.split("<br
|
| 342 |
|
| 343 |
# clean up
|
| 344 |
results = [r.replace("\n", "").replace(" ", "") for r in results]
|
| 345 |
|
| 346 |
# filter html and empty strings
|
| 347 |
results = [r for r in results if ("<" not in r) and (len(r) > 0)]
|
| 348 |
-
print(results)
|
| 349 |
|
| 350 |
return random.sample(results, min(len(results), n_words))
|
| 351 |
|
|
|
|
| 21 |
|
| 22 |
LANGUAGE = st.sidebar.radio("Language", ["english", "dutch"],0)
|
| 23 |
if LANGUAGE == "english":
|
| 24 |
+
MODEL_PATH = "bert-large-cased-whole-word-masking-finetuned-squad"
|
| 25 |
elif LANGUAGE == "dutch":
|
| 26 |
MODEL_PATH = "GroNLP/bert-base-dutch-cased"
|
| 27 |
else:
|
|
|
|
| 305 |
if language == "english":
|
| 306 |
return query_datamuse_api(last_word, n_rhymes)
|
| 307 |
elif language == "dutch":
|
|
|
|
| 308 |
return mick_rijmwoordenboek(last_word, n_rhymes)
|
| 309 |
else:
|
| 310 |
raise NotImplementedError(f"Unsupported language ({language}) expected 'english' or 'dutch'.")
|
|
|
|
| 337 |
html = get(url)
|
| 338 |
soup = Soup(html)
|
| 339 |
|
| 340 |
+
results = soup.find("div", {"id": "rhymeResultsWords"}).html.split("<br />")
|
| 341 |
|
| 342 |
# clean up
|
| 343 |
results = [r.replace("\n", "").replace(" ", "") for r in results]
|
| 344 |
|
| 345 |
# filter html and empty strings
|
| 346 |
results = [r for r in results if ("<" not in r) and (len(r) > 0)]
|
|
|
|
| 347 |
|
| 348 |
return random.sample(results, min(len(results), n_words))
|
| 349 |
|