johann22 commited on
Commit
1dccf7f
1 Parent(s): 627fa51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -17,9 +17,10 @@ agents =[
17
  "WEB_DEV",
18
  "AI_SYSTEM_PROMPT",
19
  "PYTHON_CODE_DEV",
20
- "CODE_REVIEW_ASSISTANT ",
21
- "CONTENT_WRITER_EDITOR ",
22
- "SOCIAL_MEDIA_MANAGER ",
 
23
  ]
24
  def generate(
25
  prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
@@ -39,6 +40,8 @@ def generate(
39
  agent = prompts.AI_SYSTEM_PROMPT
40
  if agent_name == "PYTHON_CODE_DEV":
41
  agent = prompts.PYTHON_CODE_DEV
 
 
42
  system_prompt=agent
43
  temperature = float(temperature)
44
  if temperature < 1e-2:
@@ -119,6 +122,7 @@ additional_inputs=[
119
 
120
  examples=[["Write a simple working game in HTML5", agents[0], None, None, None, None, ],
121
  ["Choose 3 useful types of AI agents, and create a detailed System Prompt to align each of them.", agents[1], None, None, None, None, ],
 
122
  ["Explain it to me in a childrens story how Nuclear Fission works", agents[4], None, None, None, None, ],
123
  ["Show a bunch of examples of catchy ways to post, 'I had a ham sandwich for lunch today'", agents[5], None, None, None, None, ],
124
  ["Write high quality personal website to show off my adventure sports hobby", agents[0], None, None, None, None, ],
 
17
  "WEB_DEV",
18
  "AI_SYSTEM_PROMPT",
19
  "PYTHON_CODE_DEV",
20
+ "CODE_REVIEW_ASSISTANT",
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,
 
40
  agent = prompts.AI_SYSTEM_PROMPT
41
  if agent_name == "PYTHON_CODE_DEV":
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:
 
122
 
123
  examples=[["Write a simple working game in HTML5", agents[0], None, None, None, None, ],
124
  ["Choose 3 useful types of AI agents, and create a detailed System Prompt to align each of them.", agents[1], None, None, None, None, ],
125
+ ["Create 3 of the funniest memes", agents[6], None, None, None, None, ],
126
  ["Explain it to me in a childrens story how Nuclear Fission works", agents[4], None, None, None, None, ],
127
  ["Show a bunch of examples of catchy ways to post, 'I had a ham sandwich for lunch today'", agents[5], None, None, None, None, ],
128
  ["Write high quality personal website to show off my adventure sports hobby", agents[0], None, None, None, None, ],