ProfRod100 commited on
Commit
ef55241
·
verified ·
1 Parent(s): 5f6bbaa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -19
app.py CHANGED
@@ -9,23 +9,21 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
9
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME).to("cpu")
10
 
11
 
12
- def gerar_resposta(messages):
13
  """
14
- Novo formato do Gradio (type='messages'):
15
- messages = [
16
- {"role": "user" | "assistant", "content": "texto"},
17
- ...
18
- ]
19
-
20
- A função deve receber apenas 'messages' e retornar uma string
21
- com a resposta do bot.
22
  """
23
 
24
  # Prompt de sistema: define o "personagem" do bot
25
  system_prefix = (
26
  "Você é o Professor DL, um professor de Deep Learning. "
27
  "Responda SEMPRE em português do Brasil, de forma simples, didática e objetiva, "
28
- "usando exemplos práticos quando possível. "
29
  "Explique conceitos como redes neurais, CNN, RNN, overfitting, regularização, etc., "
30
  "sem fórmulas muito pesadas."
31
  )
@@ -33,13 +31,14 @@ def gerar_resposta(messages):
33
  # Monta o texto de conversa
34
  texto = system_prefix + "\n\n"
35
 
36
- for msg in messages:
37
- role = msg.get("role", "user")
38
- content = msg.get("content", "")
39
- if role == "user":
40
- texto += f"Aluno: {content}\n"
41
- else:
42
- texto += f"Professor DL: {content}\n"
 
43
 
44
  # Última fala deve ser do professor
45
  texto += "Professor DL:"
@@ -83,7 +82,7 @@ def gerar_resposta(messages):
83
  descricao = """
84
  ### 🤖 Professor DL – Chatbot de Deep Learning
85
 
86
- Este chatbot foi treinado para atuar como um **professor de Deep Learning**:
87
 
88
  - Explica conceitos de redes neurais profundas
89
  - Fala sobre CNN, RNN, overfitting, regularização, etc.
@@ -105,8 +104,8 @@ iface = gr.ChatInterface(
105
  "O que é overfitting?",
106
  "Como começo a estudar Deep Learning na prática?",
107
  ],
 
108
  )
109
 
110
  if __name__ == "__main__":
111
  iface.launch()
112
-
 
9
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME).to("cpu")
10
 
11
 
12
+ def gerar_resposta(messages, history=None):
13
  """
14
+ Gradio (type='messages') chama a função com:
15
+ - messages: lista de mensagens no formato [{"role": ..., "content": ...}, ...]
16
+ - history: (segundo argumento, que não vamos usar)
17
+
18
+ Por isso deixamos history=None para evitar o TypeError.
19
+ A função deve retornar APENAS uma string com a resposta.
 
 
20
  """
21
 
22
  # Prompt de sistema: define o "personagem" do bot
23
  system_prefix = (
24
  "Você é o Professor DL, um professor de Deep Learning. "
25
  "Responda SEMPRE em português do Brasil, de forma simples, didática e objetiva, "
26
+ 'chamando a outra pessoa de "você" e usando exemplos práticos quando possível. '
27
  "Explique conceitos como redes neurais, CNN, RNN, overfitting, regularização, etc., "
28
  "sem fórmulas muito pesadas."
29
  )
 
31
  # Monta o texto de conversa
32
  texto = system_prefix + "\n\n"
33
 
34
+ if messages:
35
+ for msg in messages:
36
+ role = msg.get("role", "user")
37
+ content = msg.get("content", "")
38
+ if role == "user":
39
+ texto += f"Aluno: {content}\n"
40
+ else:
41
+ texto += f"Professor DL: {content}\n"
42
 
43
  # Última fala deve ser do professor
44
  texto += "Professor DL:"
 
82
  descricao = """
83
  ### 🤖 Professor DL – Chatbot de Deep Learning
84
 
85
+ Este chatbot atua como um **professor de Deep Learning**:
86
 
87
  - Explica conceitos de redes neurais profundas
88
  - Fala sobre CNN, RNN, overfitting, regularização, etc.
 
104
  "O que é overfitting?",
105
  "Como começo a estudar Deep Learning na prática?",
106
  ],
107
+ cache_examples=False, # evita rodar exemplos no startup e quebrar o app
108
  )
109
 
110
  if __name__ == "__main__":
111
  iface.launch()