Spaces:
Running
Running
Update app.py
Browse files
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, ],
|