Merge branch 'main' of https://huggingface.co/spaces/Jamal811/Test-Chatbot

app.py

@@ -1,17 +1,19 @@
-from transformers import pipeline, Conversation
+from transformers import pipeline
 import gradio as gr
 
-chatbot = pipeline(model="facebook/blenderbot-400M-distill")
-
-message_list = []
-response_list = []
+# Use the BlenderBot model with the 'text2text-generation' task
+chatbot = pipeline("text2text-generation", model="facebook/blenderbot-400M-distill")
 
+# Function to handle conversation
 def vanilla_chatbot(message, history):
-    conversation = Conversation(text=message, past_user_inputs=message_list, generated_responses=response_list)
-    conversation = chatbot(conversation)
-
-    return conversation.generated_responses[-1]
+    # Generate a response using the chatbot model
+    response = chatbot(message)
+    
+    # Return the generated response
+    return response[0]['generated_text']
 
-demo_chatbot = gr.ChatInterface(vanilla_chatbot, title="Vanilla Chatbot", description="Enter text to start chatting.")
+# Initialize the Gradio ChatInterface with 'messages' format
+demo_chatbot = gr.ChatInterface(vanilla_chatbot, title="Vanilla Chatbot", description="Enter text to start chatting.", type="messages")
 
-demo_chatbot.launch()
+# Launch the Gradio interface with a public link
+demo_chatbot.launch(share=True)