bourdoiscatie commited on
Commit
d619933
·
1 Parent(s): 9935db0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
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]: