johann22 commited on
Commit
809cb4d
1 Parent(s): bb78c41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -21,6 +21,7 @@ agents =[
21
  "CONTENT_WRITER_EDITOR",
22
  "SOCIAL_MEDIA_MANAGER",
23
  "MEME_GENERATOR",
 
24
  ]
25
  def generate(
26
  prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
@@ -42,6 +43,8 @@ def generate(
42
  agent = prompts.PYTHON_CODE_DEV
43
  if agent_name == "MEME_GENERATOR":
44
  agent = prompts.MEME_GENERATOR
 
 
45
  system_prompt=agent
46
  temperature = float(temperature)
47
  if temperature < 1e-2:
 
21
  "CONTENT_WRITER_EDITOR",
22
  "SOCIAL_MEDIA_MANAGER",
23
  "MEME_GENERATOR",
24
+ "QUESTION_GENERATOR",
25
  ]
26
  def generate(
27
  prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
 
43
  agent = prompts.PYTHON_CODE_DEV
44
  if agent_name == "MEME_GENERATOR":
45
  agent = prompts.MEME_GENERATOR
46
+ if agent_name == "QUESTION_GENERATOR":
47
+ agent = prompts.QUESTION_GENERATOR
48
  system_prompt=agent
49
  temperature = float(temperature)
50
  if temperature < 1e-2: