TurtleLiu commited on
Commit
c2cedec
·
1 Parent(s): b8d1493

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -1,3 +1,8 @@
 
 
 
 
 
1
  from huggingface_hub import InferenceClient
2
  import gradio as gr
3
 
@@ -15,7 +20,7 @@ def format_prompt(message, history):
15
  return prompt
16
 
17
 
18
- '''
19
  def format_prompt(message, history):
20
  prompt = "<s>"
21
  for user_prompt, bot_response in history:
@@ -23,7 +28,7 @@ def format_prompt(message, history):
23
  prompt += f" {bot_response}</s> "
24
  prompt += f"[INST] As a psychology counselor assistant, provide an assessment and plan for the following counseling notes. Please present a summary, don't make it so long. Present in lines.: {message} [/INST]"
25
  return prompt
26
- '''
27
 
28
  def generate(
29
  prompt, history, temperature=0.9, max_new_tokens=1024, top_p=0.95, repetition_penalty=1.0,
@@ -65,4 +70,5 @@ gr.ChatInterface(
65
  description="Enter counseling notes to generate an assessment and plan.",
66
  examples=examples,
67
  concurrency_limit=20,
68
- ).launch(show_api=True, debug=True)
 
 
1
+ import gradio as gr
2
+
3
+ gr.load("models/TurtleLiu/mistral7b_psychology_bot").launch()
4
+
5
+ '''
6
  from huggingface_hub import InferenceClient
7
  import gradio as gr
8
 
 
20
  return prompt
21
 
22
 
23
+
24
  def format_prompt(message, history):
25
  prompt = "<s>"
26
  for user_prompt, bot_response in history:
 
28
  prompt += f" {bot_response}</s> "
29
  prompt += f"[INST] As a psychology counselor assistant, provide an assessment and plan for the following counseling notes. Please present a summary, don't make it so long. Present in lines.: {message} [/INST]"
30
  return prompt
31
+
32
 
33
  def generate(
34
  prompt, history, temperature=0.9, max_new_tokens=1024, top_p=0.95, repetition_penalty=1.0,
 
70
  description="Enter counseling notes to generate an assessment and plan.",
71
  examples=examples,
72
  concurrency_limit=20,
73
+ ).launch(show_api=False, debug=True)
74
+ '''