Spaces:
Running
Running
Update app.py
Browse files
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="
|
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")
|