shekhardhangar commited on
Commit
afba2c3
Β·
verified Β·
1 Parent(s): 235e0fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from threading import Thread
8
 
9
  warnings.filterwarnings("ignore")
10
 
11
- PEFT_MODEL = "givyboy/phi-2-finetuned-mental-health-conversational"
12
 
13
  SYSTEM_PROMPT = """Answer the following question truthfully.
14
  If you don't know the answer, respond 'Sorry, I don't know the answer to this question.'.
 
8
 
9
  warnings.filterwarnings("ignore")
10
 
11
+ PEFT_MODEL = "givyboy/TinyLlama-1.1B-Chat-v1.0-mental-health-conversational"
12
 
13
  SYSTEM_PROMPT = """Answer the following question truthfully.
14
  If you don't know the answer, respond 'Sorry, I don't know the answer to this question.'.