Update app.py

app.py

@@ -56,6 +56,8 @@ dataset = load_counseling_dataset()
 def load_text_generation_model():
     return pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B")
 
+text_generator = load_text_generation_model()
+
 # Streamlit app
 st.title("Mental Health Counseling Chat")
 st.markdown("""