Spaces:
Runtime error
Runtime error
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")
|