Update app.py

app.py

@@ -4,10 +4,17 @@ from transformers import pipeline
 # Load the conversational pipeline with the "facebook/blenderbot-400M-distill" model
 pipe = pipeline("conversational", model="facebook/blenderbot-400M-distill")
 
-def chat_with_model(conversation):
-    conversation = [{"role": "system", "content": "You are a helpful assistant."}]
+def chat_with_model(input_text):
+    # Start a new conversation with the user's input
+    conversation = [{"role": "user", "content": input_text}]
+    
+    # Generate a response from the model
     response = pipe(conversation)
-    return response[0]['message']['content']
+    
+    # Extract the assistant's reply
+    reply = response[0]['message']['content']
+    
+    return reply
 
 iface = gr.Interface(
     fn=chat_with_model,