Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 15 |
-
messages
|
| 16 |
-
|
| 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 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
|
|
|
| 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
|
| 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()
|
|
|