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 |
|