Update app.py

app.py

@@ -11,8 +11,12 @@ st.title("Hugging Face Conversational Model Demo")
 user_message = st.text_input("You:", "")
 
 if st.button("Send"):
+    # Format the conversation for the conversational pipeline
+    conversation_history = [{"role": "system", "content": "You are a helpful assistant."},
+                           {"role": "user", "content": user_message}]
+
     # Get the model's response
-    model_response = conversational_pipeline(user_message)[0]['generated_responses'][0]
+    model_response = conversational_pipeline(conversation_history)[0]['generated_responses'][0]
 
     # Display the model's response
     st.text_area("Model:", model_response, height=100)