alex-abb commited on
Commit
c7c5ea0
·
verified ·
1 Parent(s): 3b14f3e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -25,15 +25,15 @@ def generate_response(user_message, history):
25
 
26
  # Préparer les entrées pour le modèle
27
  conversation = Conversation()
28
- for message in messages:
29
- if message["role"] == "user":
30
- conversation.add_user_input(message["content"])
31
- else:
32
- conversation.append_response(message["content"])
33
-
34
  # Générer une réponse
35
  result = pipe(conversation)
36
- response = result[-1]['generated_text']
37
 
38
  # Mettre à jour l'historique
39
  history.append((user_message, response))
 
25
 
26
  # Préparer les entrées pour le modèle
27
  conversation = Conversation()
28
+ for past_user_input, past_bot_response in history:
29
+ conversation.add_user_input(past_user_input)
30
+ conversation.append_response(past_bot_response)
31
+
32
+ conversation.add_user_input(user_message)
33
+
34
  # Générer une réponse
35
  result = pipe(conversation)
36
+ response = result[0]['generated_text']
37
 
38
  # Mettre à jour l'historique
39
  history.append((user_message, response))