Update app.py

app.py

@@ -45,15 +45,14 @@ from transformers import pipeline
 # Configure the Hugging Face API key
 HUGGINGFACE_API_KEY = st.secrets['huggingface_api_key']
 
-# Initialize the Hugging Face conversational model
-# You can replace 'microsoft/DialoGPT-medium' with any other supported conversational model
-chatbot = pipeline("conversational", model="microsoft/DialoGPT-medium")
+# Initialize the Hugging Face text-generation model (DialoGPT or other conversational models)
+chatbot = pipeline("text-generation", model="microsoft/DialoGPT-medium", api_key=HUGGINGFACE_API_KEY)
 
 # Function to get response from the Hugging Face model
 def get_chatbot_response(user_input):
     try:
-        conversation = chatbot(user_input)
-        return conversation[0]['generated_text']  # Extract the generated response
+        response = chatbot(user_input, max_length=1000, pad_token_id=50256)  # Generate response
+        return response[0]['generated_text']  # Extract the generated response
     except Exception as e:
         return f"Error: {str(e)}"
 
@@ -125,4 +124,3 @@ if st.session_state["history"]:
         st.markdown(f'<div class="chat-bubble user-bubble"><span class="emoji">👤</span>You: {user_input}</div>', unsafe_allow_html=True)
         st.markdown(f'<div class="chat-bubble bot-bubble"><span class="emoji">🤖</span>Bot: {response}</div>', unsafe_allow_html=True)
     st.markdown('</div>', unsafe_allow_html=True)
-