Update app.py

app.py

@@ -14,7 +14,7 @@ image = Image.open(image_path)
 st.sidebar.image(image, caption=None, width=None, use_column_width=None, clamp=False, channels="RGB", output_format="auto")
 
 @st.cache_resource
-def load_model(version):
+def load_model(version,text):
     if version == "version base":
         ner = pipeline('token-classification', model='bourdoiscatie/Camembert-NER-base-frenchNER', tokenizer='bourdoiscatie/Camembert-NER-base-frenchNER', grouped_entities=True)
         result = ner(text)
@@ -62,8 +62,6 @@ def color_annotation(to_print,transcript) :
             
 st.markdown("<h2 style='text-align: center'>Camembert-NER", unsafe_allow_html=True)    
 st.markdown("<h4 style='text-align: center'>"+version, unsafe_allow_html=True)    
-model = load_model(version)
-
 option = st.selectbox(
     'Choix du mode',
     ('Texte libre', 'Exemple'))
@@ -71,7 +69,7 @@ option = st.selectbox(
 if option == "Exemple":
     text = st.text_area("Votre texte", value="Assurés de disputer l'Euro 2024 en Allemagne l'été prochain (du 14 juin au 14 juillet) depuis leur victoire aux Pays-Bas, les Bleus ont fait le nécessaire pour avoir des certitudes. Avec six victoires en six matchs officiels et un seul but encaissé, Didier Deschamps a consolidé les acquis de la dernière Coupe du monde. Les joueurs clés sont connus : Kylian Mbappé, Aurélien Tchouameni, Antoine Griezmann, Ibrahima Konaté ou encore Mike Maignan.",height=175)
     if text:
-        display = color_annotation(load_model(text),text)
+        display = color_annotation(load_model(version,text),text)
         st.write(display)
         list_to_display = [] # pour pouvoir afficher la couleur, on doit passer les mots à colorier de str en tuple
         for i in range(len(display.split('","'))):
@@ -90,7 +88,7 @@ else:
     if text:
         col1, col2, col3 = st.columns(3)
         if col2.button('Appliquer le modèle'):
-            display = color_annotation(load_model(text),text)
+            display = color_annotation(load_model(version,text),text)
             list_to_display = [] # pour pouvoir afficher la couleur, on doit passer les mots à colorier de str en tuple
             for i in range(len(display.split('","'))):
                 if "(" in display.split('","')[i]: