ShieldX commited on
Commit
9efacf0
β€’
1 Parent(s): 54cc37c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ model = model.to('cpu')
8
 
9
  title = "🌱 ManoVyadh 🌱"
10
  description = "Mental Health Counselling Chatbot"
11
- examples = [["I am feeling sad for my friend's divorce"]]
12
 
13
  def predict(message, history):
14
  def formatted_prompt(question)-> str:
@@ -34,7 +34,7 @@ def predict(message, history):
34
  top_k=5,
35
  temperature=0.5,
36
  repetition_penalty=1.2,
37
- max_new_tokens=128,
38
  eos_token_id=tokenizer.eos_token_id,
39
  pad_token_id=tokenizer.eos_token_id
40
  )
 
8
 
9
  title = "🌱 ManoVyadh 🌱"
10
  description = "Mental Health Counselling Chatbot"
11
+ examples = [["I have been feeling more and more down for over a month. I have started having trouble sleeping due to panic attacks, but they are almost never triggered by something that I know of.", "I self-harm, and I stop for awhile. Then when I see something sad or depressing, I automatically want to self-harm.", "I am feeling sad for my friend's divorce"]]
12
 
13
  def predict(message, history):
14
  def formatted_prompt(question)-> str:
 
34
  top_k=5,
35
  temperature=0.5,
36
  repetition_penalty=1.2,
37
+ max_new_tokens=64,
38
  eos_token_id=tokenizer.eos_token_id,
39
  pad_token_id=tokenizer.eos_token_id
40
  )