Update app.py

app.py

@@ -1,4 +1,4 @@
-from transformers import pipeline, Conversation
+from transformers import pipeline
 import gradio as gr
 import os
 import spaces
@@ -16,18 +16,17 @@ def generate_response(user_message, history):
     # Ajouter le message système au début de l'historique des messages
     messages = [{"role": "system", "content": system_message}] + [{"role": "user", "content": user_message}]
     
-    # Créer une conversation avec l'historique des messages
-    conversation = Conversation()
+    # Créer une chaîne de caractères représentant la conversation
+    conversation_text = system_message + "\n"
     for past_user_input, past_bot_response in history:
-        conversation.add_user_input(past_user_input)
-        conversation.append_response(past_bot_response)
+        conversation_text += f"User: {past_user_input}\nAssistant: {past_bot_response}\n"
     
     # Ajouter le nouveau message de l'utilisateur
-    conversation.add_user_input(user_message)
+    conversation_text += f"User: {user_message}\n"
     
     # Générer une réponse
-    result = pipe(conversation)
-    response = result[0]['generated_text']
+    result = pipe(conversation_text, max_new_tokens=150)
+    response = result[0]['generated_text'].split("User: ")[-1].strip()  # Extraire la réponse générée
     
     # Mettre à jour l'historique
     history.append((user_message, response))
@@ -41,5 +40,4 @@ iface = gr.Interface(
     live=True,
 )
 
-gr.ChatInterface(generate_response).launch()
-
+iface.launch()