Update app.py

app.py

@@ -1,11 +1,19 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 
-# Configure the Hugging Face API key
+# Configure the Hugging Face API key (no need to pass it in the pipeline call)
 HF_API_KEY = st.secrets['huggingface_api_key']
 
-# Initialize the Hugging Face model with the API key (authentication done separately)
-generator = pipeline('text-generation', model='gpt2', tokenizer='gpt2')
+# Ensure you're logged in using the Hugging Face CLI if using private models
+# huggingface-cli login
+
+# Initialize the Hugging Face model and tokenizer
+model_name = 'gpt2-medium'  # or another GPT-2 version you want to use
+model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+# Initialize the text generation pipeline
+generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
 
 # Function to get response from the Hugging Face model
 def get_chatbot_response(user_input):