hgrif commited on
Commit
462d4c0
β€’
1 Parent(s): 6a177e5

Up iters for NL

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,14 +12,15 @@ from rhyme_with_ai.rhyme_generator import RhymeGenerator
12
 
13
  DEFAULT_QUERY = "Machines will take over the world soon"
14
  N_RHYMES = 10
15
- ITER_FACTOR = 5
16
 
17
 
18
  LANGUAGE = st.sidebar.radio("Language", ["english", "dutch"],0)
19
  if LANGUAGE == "english":
20
  MODEL_PATH = "bert-large-cased-whole-word-masking"
 
21
  elif LANGUAGE == "dutch":
22
  MODEL_PATH = "GroNLP/bert-base-dutch-cased"
 
23
  else:
24
  raise NotImplementedError(f"Unsupported language ({LANGUAGE}) expected 'english' or 'dutch'.")
25
 
 
12
 
13
  DEFAULT_QUERY = "Machines will take over the world soon"
14
  N_RHYMES = 10
 
15
 
16
 
17
  LANGUAGE = st.sidebar.radio("Language", ["english", "dutch"],0)
18
  if LANGUAGE == "english":
19
  MODEL_PATH = "bert-large-cased-whole-word-masking"
20
+ ITER_FACTOR = 5
21
  elif LANGUAGE == "dutch":
22
  MODEL_PATH = "GroNLP/bert-base-dutch-cased"
23
+ ITER_FACTOR = 10 # Faster model
24
  else:
25
  raise NotImplementedError(f"Unsupported language ({LANGUAGE}) expected 'english' or 'dutch'.")
26