Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,33 +1,33 @@
|
|
1 |
-
# Importa
|
2 |
-
import gradio as gr
|
3 |
-
from transformers import pipeline
|
4 |
|
5 |
-
#
|
6 |
def mostrar_resultado(texto):
|
7 |
-
#
|
8 |
if len(texto) < 400:
|
9 |
return 'O texto precisa ter no mínimo 400 caracteres'
|
10 |
|
11 |
-
#
|
12 |
-
nota = int(classificador(texto)[0]['label']) *
|
13 |
return nota
|
14 |
|
15 |
-
# Inicializa o pipeline de classificação de texto com o modelo
|
16 |
classificador = pipeline(
|
17 |
-
'text-classification',
|
18 |
-
model='adrianoL/distilbert-pt-cased-redacao-nota-modelo', #
|
19 |
-
framework='tf' #
|
20 |
)
|
21 |
|
22 |
-
#
|
23 |
app = gr.Interface(
|
24 |
-
fn=mostrar_resultado,
|
25 |
-
inputs=gr.Textbox(label='Digite o texto'), #
|
26 |
-
outputs=gr.Textbox(label='Nota da redação'),
|
27 |
-
title='Insira um texto para receber uma nota entre 0 e 10'
|
28 |
)
|
29 |
|
30 |
-
#
|
31 |
if __name__ == "__main__":
|
32 |
app.launch(share=True)
|
33 |
|
|
|
1 |
+
# Importa as bibliotecas necessárias
|
2 |
+
import gradio as gr # Para criação da interface gráfica interativa
|
3 |
+
from transformers import pipeline # Para carregar o modelo de classificação de texto
|
4 |
|
5 |
+
# Função para processar o texto e retornar a nota da redação
|
6 |
def mostrar_resultado(texto):
|
7 |
+
# Valida se o texto tem o número mínimo de caracteres
|
8 |
if len(texto) < 400:
|
9 |
return 'O texto precisa ter no mínimo 400 caracteres'
|
10 |
|
11 |
+
# Executa a classificação do texto e converte o rótulo em uma nota de 0 a 10
|
12 |
+
nota = int(classificador(texto)[0]['label']) * 1
|
13 |
return nota
|
14 |
|
15 |
+
# Inicializa o pipeline de classificação de texto com o modelo personalizado no Hugging Face
|
16 |
classificador = pipeline(
|
17 |
+
task='text-classification', # Define a tarefa como classificação de texto
|
18 |
+
model='adrianoL/distilbert-pt-cased-redacao-nota-modelo', # Especifica o modelo customizado
|
19 |
+
framework='tf' # Utiliza o TensorFlow como backend
|
20 |
)
|
21 |
|
22 |
+
# Configura a interface gráfica com Gradio
|
23 |
app = gr.Interface(
|
24 |
+
fn=mostrar_resultado, # Função chamada ao submeter o texto
|
25 |
+
inputs=gr.Textbox(label='Digite o texto'), # Caixa de texto para entrada do usuário
|
26 |
+
outputs=gr.Textbox(label='Nota da redação'), # Caixa de texto para exibir a nota resultante
|
27 |
+
title='Insira um texto para receber uma nota entre 0 e 10' # Título da interface
|
28 |
)
|
29 |
|
30 |
+
# Executa a aplicação localmente com opção de compartilhamento
|
31 |
if __name__ == "__main__":
|
32 |
app.launch(share=True)
|
33 |
|