Update space

app.py

@@ -1,4 +1,3 @@
-# This is my first chatbot example
 from transformers import pipeline, Conversation
 import gradio as gr
 
@@ -7,12 +6,11 @@ chatbot = pipeline(model="facebook/blenderbot-400M-distill", task="conversationa
 message_list = []
 response_list = []
 
-
 def vanilla_chatbot(message, history):
     conversation = Conversation(text=message, past_user_inputs=message_list, generated_responses=response_list)
     conversation = chatbot(conversation)
 
-    return conversation[0].generated_responses[-1]
+    return conversation.generated_responses[-1]
 
 demo_chatbot = gr.ChatInterface(vanilla_chatbot, title="Vanilla Chatbot", description="Enter text to start chatting.")