Update app.py

app.py

@@ -6,7 +6,7 @@ import re
 
 st.sidebar.header("**Instructions**")
 st.sidebar.markdown("Démonstrateur du modèle [Camembert-NER](CATIE-AQ/Camembert-NER-base-frenchNER) entraîné sur plus de 425 000 données en français. Le modèle est capable d'étiquetter les entités LOC (Localisation), PER (Personne), ORG (Organisation) dans le texte fourni. Il est disponible en version *base* (110M de paramètres) et *large* (335M de paramètres). Pour l'essayer, sélectionnez la version de votre choix ci-dessous, puis renseignez un texte. Enfin appuyez sur le bouton « Appliquer le modèle » pour observer la réponse trouvée par le modèle. Pour en savoir plus sur ce modèle, vous pouvez lire l'[article de blog](https://blog.vaniila.ai/NER/) détaillant la démarche suvie.")
-version = st.sidebar.radio("Choix de la version du modèle :", ["version base", "version large"])
+version = st.sidebar.radio("Choix de la version du modèle :", ["version 3 entités", "version 4 entités"])
 st.sidebar.markdown("---")
 st.sidebar.markdown("Ce modèle a été entraîné via la plateforme [*Vaniila*](https://www.vaniila.ai/) du [CATIE](https://www.catie.fr/).")
 
@@ -71,24 +71,6 @@ 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:
         to_print = load_model(version,text)
-        
-        dict_to_del = []
-        for i in range(len(to_print)-1):
-            if to_print[i]["end"] == to_print[i+1]["start"]:
-                to_print[i+1]["word"] = to_print[i]["word"]+to_print[i+1]["word"]
-                to_print[i+1]["score"] = (to_print[i]["score"]+to_print[i+1]["score"])/2
-                to_print[i+1]["start"] = to_print[i]["start"]
-                dict_to_del.append(i)
-        to_print = [j for i, j in enumerate(to_print) if i not in dict_to_del]
-        dict_to_del = []
-        for i in range(len(to_print)-1):
-            if (to_print[i]["end"] == to_print[i+1]["start"]-1):
-                to_print[i+1]["word"] = to_print[i]["word"]+" "+to_print[i+1]["word"]
-                to_print[i+1]["score"] = (to_print[i]["score"]+to_print[i+1]["score"])/2
-                to_print[i+1]["start"] = to_print[i]["start"]
-                dict_to_del.append(i)
-        to_print = [j for i, j in enumerate(to_print) if i not in dict_to_del]
-
         display = color_annotation(to_print,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('","'))):