johann22 commited on
Commit
0be2715
1 Parent(s): 9cc5276

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,7 +6,6 @@ client = InferenceClient(
6
  )
7
 
8
 
9
-
10
  def format_prompt(message, history):
11
  prompt = "<s>"
12
  for user_prompt, bot_response in history:
@@ -20,7 +19,7 @@ agents =[
20
  "PYTHON_CODE_DEV"
21
  ]
22
  def generate(
23
- prompt, history, sys_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, agent_name=agents[0],
24
  ):
25
  agent=prompts.WEB_DEV
26
  if agent_name == "WEB_DEV":
 
6
  )
7
 
8
 
 
9
  def format_prompt(message, history):
10
  prompt = "<s>"
11
  for user_prompt, bot_response in history:
 
19
  "PYTHON_CODE_DEV"
20
  ]
21
  def generate(
22
+ prompt, history, agent_name=agents[0], sys_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
23
  ):
24
  agent=prompts.WEB_DEV
25
  if agent_name == "WEB_DEV":