fix: language

app.py

@@ -16,7 +16,7 @@ DEFAULT_QUERY = "Machines will take over the world soon"
 N_RHYMES = 10
 
 
-"""LANGUAGE = st.sidebar.radio("Language", ["english", "dutch", "french"],0)
+LANGUAGE = st.sidebar.radio("Language", ["english", "dutch", "french"],0)
 if LANGUAGE == "english":
     MODEL_PATH = "bert-large-cased-whole-word-masking"
     ITER_FACTOR = 5
@@ -24,14 +24,15 @@ elif LANGUAGE == "dutch":
     MODEL_PATH = "GroNLP/bert-base-dutch-cased"
     ITER_FACTOR = 10  # Faster model
 elif LANGUAGE == "french":
-
+    MODEL_PATH = "camembert-base"
+    ITER_FACTOR = 5
 else:
     raise NotImplementedError(f"Unsupported language ({LANGUAGE}) expected 'english','dutch' or 'french.")
-"""
 
-LANGUAGE = "french"
+
+"""LANGUAGE = "french"
 MODEL_PATH = "camembert-base"
-ITER_FACTOR = 5
+ITER_FACTOR = 5"""
 
 def main():
     st.markdown(
@@ -100,7 +101,7 @@ def load_model(model_path, language):
         tokenizer = CamembertTokenizer(vocab_file='rhyme_with_ai/dict.txt')
         return (
         CamembertModel.from_pretrained(model_path),
-        CamembertTokenizer.from_pretrained(model_path),
+        tokenizer.from_pretrained(model_path),
     )
 
 def display_output(status_text, query, current_sentences, previous_sentences):