SlickSlick commited on
Commit
3003d01
·
verified ·
1 Parent(s): 585e195

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -58
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()