Divymakesml commited on
Commit
9f937db
·
verified ·
1 Parent(s): 842983b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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-small-summarization",
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="SafeSpace: AI Compassionate Listener",
93
  page_icon="🧠"
94
  )
95
 
96
- st.title("🧠 SafeSpace: Compassionate AI Listener")
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