flaviaggp commited on
Commit
d07b7c0
·
1 Parent(s): 43f0830

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +37 -0
app.py ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # https://medium.com/@groxli/create-a-spacy-visualizer-with-streamlit-8b9b41b36745
2
+ # https://gitlab.com/groxli/spacy-visualizer-with-streamlit/-/blob/main/visualizer.py
3
+
4
+ import streamlit as st
5
+ import spacy
6
+ from spacy import displacy
7
+
8
+ st.title("Web APP LeNER-Br")
9
+
10
+ #Texto:
11
+ 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.')
12
+
13
+ # Função que carrega os modelos:
14
+ def load_models():
15
+ sm_model = spacy.load("modelo_lener_sm")
16
+ lg_model = spacy.load("modelo_lener_lg")
17
+ models = {"sm": sm_model, "lg": lg_model}
18
+ return models
19
+
20
+ models = load_models()
21
+ selected_type = st.sidebar.selectbox("Selecione o tipo do modelo", options=["sm", "lg"]) # caixa de seleção do modelo
22
+ selected_model = models[selected_type]
23
+ doc= selected_model(input_text) # função doc que processa o texto de acordo com a opção escolhida acima
24
+
25
+ # Cabeçalho
26
+ st.header("Visualizador de entidades")
27
+
28
+ #Cores:
29
+ colors = {"LEGISLACAO": "linear-gradient(90deg, #aa9cfc, #fc9ce7)", 'JURISPRUDENCIA': "#ccfbf1", 'LOCAL': "#ffedd5", 'ORGANIZACAO': "#fae8ff", 'PESSOA': "#e0f2fe", 'TEMPO': "#fefde0", }
30
+ options = {"ents": ["LEGISLACAO", "JURISPRUDENCIA", "LOCAL", "ORGANIZACAO", "PESSOA", "TEMPO",], "colors": colors}
31
+
32
+ #Html:
33
+ ent_html = displacy.render(doc, style="ent", options=options, jupyter=False) # https://spacy.io/usage/visualizers
34
+
35
+ st.markdown(ent_html, unsafe_allow_html=True)
36
+
37
+ # streamlit run render.py