Spaces:
Running
Running
Commit
·
d619933
1
Parent(s):
9935db0
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ image = Image.open(image_path)
|
|
14 |
st.sidebar.image(image, caption=None, width=None, use_column_width=None, clamp=False, channels="RGB", output_format="auto")
|
15 |
|
16 |
@st.cache_resource
|
17 |
-
def load_model(version):
|
18 |
if version == "version base":
|
19 |
ner = pipeline('token-classification', model='bourdoiscatie/Camembert-NER-base-frenchNER', tokenizer='bourdoiscatie/Camembert-NER-base-frenchNER', grouped_entities=True)
|
20 |
result = ner(text)
|
@@ -62,8 +62,6 @@ def color_annotation(to_print,transcript) :
|
|
62 |
|
63 |
st.markdown("<h2 style='text-align: center'>Camembert-NER", unsafe_allow_html=True)
|
64 |
st.markdown("<h4 style='text-align: center'>"+version, unsafe_allow_html=True)
|
65 |
-
model = load_model(version)
|
66 |
-
|
67 |
option = st.selectbox(
|
68 |
'Choix du mode',
|
69 |
('Texte libre', 'Exemple'))
|
@@ -71,7 +69,7 @@ option = st.selectbox(
|
|
71 |
if option == "Exemple":
|
72 |
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)
|
73 |
if text:
|
74 |
-
display = color_annotation(load_model(text),text)
|
75 |
st.write(display)
|
76 |
list_to_display = [] # pour pouvoir afficher la couleur, on doit passer les mots à colorier de str en tuple
|
77 |
for i in range(len(display.split('","'))):
|
@@ -90,7 +88,7 @@ else:
|
|
90 |
if text:
|
91 |
col1, col2, col3 = st.columns(3)
|
92 |
if col2.button('Appliquer le modèle'):
|
93 |
-
display = color_annotation(load_model(text),text)
|
94 |
list_to_display = [] # pour pouvoir afficher la couleur, on doit passer les mots à colorier de str en tuple
|
95 |
for i in range(len(display.split('","'))):
|
96 |
if "(" in display.split('","')[i]:
|
|
|
14 |
st.sidebar.image(image, caption=None, width=None, use_column_width=None, clamp=False, channels="RGB", output_format="auto")
|
15 |
|
16 |
@st.cache_resource
|
17 |
+
def load_model(version,text):
|
18 |
if version == "version base":
|
19 |
ner = pipeline('token-classification', model='bourdoiscatie/Camembert-NER-base-frenchNER', tokenizer='bourdoiscatie/Camembert-NER-base-frenchNER', grouped_entities=True)
|
20 |
result = ner(text)
|
|
|
62 |
|
63 |
st.markdown("<h2 style='text-align: center'>Camembert-NER", unsafe_allow_html=True)
|
64 |
st.markdown("<h4 style='text-align: center'>"+version, unsafe_allow_html=True)
|
|
|
|
|
65 |
option = st.selectbox(
|
66 |
'Choix du mode',
|
67 |
('Texte libre', 'Exemple'))
|
|
|
69 |
if option == "Exemple":
|
70 |
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)
|
71 |
if text:
|
72 |
+
display = color_annotation(load_model(version,text),text)
|
73 |
st.write(display)
|
74 |
list_to_display = [] # pour pouvoir afficher la couleur, on doit passer les mots à colorier de str en tuple
|
75 |
for i in range(len(display.split('","'))):
|
|
|
88 |
if text:
|
89 |
col1, col2, col3 = st.columns(3)
|
90 |
if col2.button('Appliquer le modèle'):
|
91 |
+
display = color_annotation(load_model(version,text),text)
|
92 |
list_to_display = [] # pour pouvoir afficher la couleur, on doit passer les mots à colorier de str en tuple
|
93 |
for i in range(len(display.split('","'))):
|
94 |
if "(" in display.split('","')[i]:
|