Spaces:
Runtime error
Runtime error
Delete app.py
Browse files
app.py
DELETED
@@ -1,58 +0,0 @@
|
|
1 |
-
import gradio as gr
|
2 |
-
from transformers import pipeline
|
3 |
-
|
4 |
-
# Modelos Públicos SEM necessidade de login
|
5 |
-
MODELOS = {
|
6 |
-
"curto": "pierreguillou/bert-base-cased-squad-v1.1-portuguese",
|
7 |
-
"longo": "dominguesm/portuguese-t5-small", # Modelo leve e público
|
8 |
-
"arbitro": "neuralmind/bert-base-portuguese-cased"
|
9 |
-
}
|
10 |
-
|
11 |
-
# Configuração à prova de falhas
|
12 |
-
try:
|
13 |
-
modelo_rapido = pipeline(
|
14 |
-
"text-generation",
|
15 |
-
model=MODELOS["curto"],
|
16 |
-
max_length=50,
|
17 |
-
do_sample=False
|
18 |
-
)
|
19 |
-
|
20 |
-
modelo_detalhado = pipeline(
|
21 |
-
"text2text-generation", # Altere para text2text-generation
|
22 |
-
model=MODELOS["longo"],
|
23 |
-
max_length=100
|
24 |
-
)
|
25 |
-
|
26 |
-
arbitro = pipeline(
|
27 |
-
"text-classification",
|
28 |
-
model=MODELOS["arbitro"]
|
29 |
-
)
|
30 |
-
except Exception as e:
|
31 |
-
raise gr.Error(f"Modelo não carregado. Erro: {str(e)}")
|
32 |
-
|
33 |
-
def chatbot(prompt):
|
34 |
-
try:
|
35 |
-
prompt_pt = f"Responda em português brasileiro de forma acadêmica: {prompt}"
|
36 |
-
|
37 |
-
# Geração de respostas
|
38 |
-
resposta_curta = modelo_rapido(prompt_pt)[0]['generated_text']
|
39 |
-
resposta_longa = modelo_detalhado(prompt_pt)[0]['generated_text']
|
40 |
-
|
41 |
-
# Limpeza básica
|
42 |
-
resposta_curta = resposta_curta.split(".")[0] + "."
|
43 |
-
resposta_longa = resposta_longa.split("\n")[0]
|
44 |
-
|
45 |
-
return f"🔵 Resposta Curta:\n{resposta_curta}\n\n🟢 Resposta Detalhada:\n{resposta_longa}"
|
46 |
-
|
47 |
-
except Exception as e:
|
48 |
-
return f"Erro ao gerar resposta. Tente reformular sua pergunta."
|
49 |
-
|
50 |
-
# Interface simplificada
|
51 |
-
with gr.Blocks() as app:
|
52 |
-
gr.Markdown("## 🤖 Chatbot Acadêmico ABNT")
|
53 |
-
entrada = gr.Textbox(label="Faça sua pergunta")
|
54 |
-
saida = gr.Textbox(label="Resposta", interactive=False)
|
55 |
-
btn = gr.Button("Enviar")
|
56 |
-
btn.click(chatbot, inputs=entrada, outputs=saida)
|
57 |
-
|
58 |
-
app.launch()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|