Update app.py

app.py

@@ -3,12 +3,16 @@ import streamlit as st
 import spacy
 from spacy import displacy
 
-#Título:
+# Título:
 st.title("Web APP LeNER-Br")
 
-#Descrição:
+# Descrição:
 st.write("Este Web App foi criado para proporcionar a comparação dos modelos spaCy [sm](https://huggingface.co/flaviaggp/pt_pipeline) e [lg](https://huggingface.co/flaviaggp/pt_lg_pipeline), voltados para a tarefa de NER com o dataset LeNER-Br.")
 
+# Legenda dos modelos:
+st.write(":red[sm: pipeline mais rápido e menor, mas menos preciso.]")
+st.write(":red[lg: pipeline maior e mais lento, mas mais preciso.]")
+
 #Texto:
 input_text = st.text_input('Insira o texto a ser analisado:', 'Que causaram evidente transtorno e evidencia a má prestação de serviço, com violação ao princípio da transparência, da confiaça e da boa-fé objetiva insertos nos artigos 4º e 6º do CDC. Por todo o acima exposto, na forma do artigo 269, I do Código de Processo Civil, conhecido e apelação não promovida. (Apelação Cível 2009 01 1 075609-5 APC Relator Desembargador JAIR SOARE.) Em relação ao CONTRATO BANCÁRIO INVERSÃO DO ÔNUS DA PROVA CDC Possibilidade da inversão do ônus da prova com base no artigo 6º, VIII, do CDC Reconhecido que o cliente tem direito de postular a exibição de documentos - Possibilidade de determinação pelo juiz incidentalmente.')
 
@@ -24,18 +28,14 @@ selected_type = st.sidebar.selectbox("Selecione o tipo do modelo", options=["sm"
 selected_model = models[selected_type]
 doc= selected_model(input_text) # função doc que processa o texto de acordo com a opção escolhida acima
 
-#Descrição:
-st.write("sm: pipeline mais rápido e menor, mas menos preciso.")
-st.write("lg: pipeline maior e mais lento, mas mais preciso.")
-
 # Cabeçalho
 st.header("Visualizador de entidades")
 
-#Cores:
+# Cores:
 colors = {"LEGISLACAO": "linear-gradient(90deg, #aa9cfc, #fc9ce7)", 'JURISPRUDENCIA': "#ccfbf1", 'LOCAL': "#ffedd5", 'ORGANIZACAO': "#fae8ff", 'PESSOA': "#e0f2fe", 'TEMPO': "#fefde0", }
 options = {"ents": ["LEGISLACAO", "JURISPRUDENCIA", "LOCAL", "ORGANIZACAO", "PESSOA", "TEMPO",], "colors": colors}
 
-#Html:
+# Html:
 ent_html = displacy.render(doc, style="ent", options=options, jupyter=False) # https://spacy.io/usage/visualizers
 
 st.markdown(ent_html, unsafe_allow_html=True)