Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ agents =[
|
|
21 |
def generate(
|
22 |
prompt, history, sys_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, agent_name=agents[0],
|
23 |
):
|
24 |
-
agent=
|
25 |
if agent_name == "WEB_DEV":
|
26 |
agent = prompts.WEB_DEV
|
27 |
if agent_name == "AI_SYSTEM_PROMPT":
|
|
|
21 |
def generate(
|
22 |
prompt, history, sys_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, agent_name=agents[0],
|
23 |
):
|
24 |
+
agent=prompts.WEB_DEV
|
25 |
if agent_name == "WEB_DEV":
|
26 |
agent = prompts.WEB_DEV
|
27 |
if agent_name == "AI_SYSTEM_PROMPT":
|