Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -28,10 +28,10 @@ class AITherapistAssistant:
|
|
28 |
device=0 if torch.cuda.is_available() else -1
|
29 |
)
|
30 |
|
31 |
-
# Summarization model
|
32 |
self.summary_model = pipeline(
|
33 |
"summarization",
|
34 |
-
model="facebook/bart-
|
35 |
device=0 if torch.cuda.is_available() else -1
|
36 |
)
|
37 |
except Exception as e:
|
@@ -89,11 +89,11 @@ class AITherapistAssistant:
|
|
89 |
|
90 |
def main():
|
91 |
st.set_page_config(
|
92 |
-
page_title="
|
93 |
page_icon="🧠"
|
94 |
)
|
95 |
|
96 |
-
st.title("🧠
|
97 |
st.write("A supportive space to share your feelings safely")
|
98 |
|
99 |
# Initialize session state
|
|
|
28 |
device=0 if torch.cuda.is_available() else -1
|
29 |
)
|
30 |
|
31 |
+
# Summarization model (using a validated model)
|
32 |
self.summary_model = pipeline(
|
33 |
"summarization",
|
34 |
+
model="facebook/bart-large-cnn",
|
35 |
device=0 if torch.cuda.is_available() else -1
|
36 |
)
|
37 |
except Exception as e:
|
|
|
89 |
|
90 |
def main():
|
91 |
st.set_page_config(
|
92 |
+
page_title="TARS: Therapist Assistance and Response System",
|
93 |
page_icon="🧠"
|
94 |
)
|
95 |
|
96 |
+
st.title("🧠 TARS: Therapist Assistance and Response System")
|
97 |
st.write("A supportive space to share your feelings safely")
|
98 |
|
99 |
# Initialize session state
|