hgrif commited on
Commit
4085796
β€’
1 Parent(s): aab7a6e

Revert "Switch to model that's not finetuned"

Browse files

This reverts commit aab7a6efeb804c6f2ad019a0068a9b00cca11706.

Files changed (1) hide show
  1. app.py +2 -4
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-uncased-whole-word-masking"
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