fix: laod model

app.py

@@ -4,6 +4,7 @@ from typing import List
 
 import streamlit as st
 from transformers import BertTokenizer, TFAutoModelForMaskedLM
+from transformers import CamembertModel, CamembertTokenizer
 
 from rhyme_with_ai.utils import color_new_words, sanitize
 from rhyme_with_ai.rhyme import query_rhyme_words
@@ -70,7 +71,7 @@ def start_rhyming(query, rhyme_words_options):
 
     rhyme_words = rhyme_words_options[:N_RHYMES]
 
-    model, tokenizer = load_model(MODEL_PATH)
+    model, tokenizer = load_model(MODEL_PATH, LANGUAGE)
     sentence_generator = RhymeGenerator(model, tokenizer)
     sentence_generator.start(query, rhyme_words)
 
@@ -84,13 +85,18 @@ def start_rhyming(query, rhyme_words_options):
 
 
 @st.cache(allow_output_mutation=True)
-def load_model(model_path):
-    return (
-        TFAutoModelForMaskedLM.from_pretrained(model_path),
-        BertTokenizer.from_pretrained(model_path),
+def load_model(model_path, language):
+    if language != "french":
+        return (
+            TFAutoModelForMaskedLM.from_pretrained(model_path),
+            BertTokenizer.from_pretrained(model_path),
+        )
+    else :
+        return (
+        CamembertModel.from_pretrained(model_path),
+        CamembertTokenizer.from_pretrained(model_path),
     )
 
-
 def display_output(status_text, query, current_sentences, previous_sentences):
     print_sentences = []
     for new, old in zip(current_sentences, previous_sentences):