Update app.py
Browse files
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:
|