tahirsher commited on
Commit
bcab110
·
verified ·
1 Parent(s): d02fed4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -51,12 +51,10 @@ def load_counseling_dataset():
51
 
52
  dataset = load_counseling_dataset()
53
 
54
- # Load a Hugging Face-compatible text-generation model
55
  @st.cache_resource
56
  def load_text_generation_model():
57
- return pipeline("text-generation", model="meta-llama/Llama-2-7b-chat-hf")
58
-
59
- text_generator = load_text_generation_model()
60
 
61
  # Streamlit app
62
  st.title("Mental Health Counseling Chat")
 
51
 
52
  dataset = load_counseling_dataset()
53
 
54
+ # Load a Hugging Face-compatible text-generation model (Open Access Alternative)
55
  @st.cache_resource
56
  def load_text_generation_model():
57
+ return pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B")
 
 
58
 
59
  # Streamlit app
60
  st.title("Mental Health Counseling Chat")