Mistri

Sleeping

johann22 commited on Dec 20, 2023

Commit

0740cc4

•

1 Parent(s): d5e2616

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ agents =[
 def generate(
         prompt, history, sys_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, agent_name=agents[0],
 ):
-    agent=prompt.AI_SYSTEM_PROMPT
     if agent_name == "WEB_DEV":
         agent = prompts.WEB_DEV
     if agent_name == "AI_SYSTEM_PROMPT":

 def generate(
         prompt, history, sys_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, agent_name=agents[0],
 ):
+    agent=prompts.WEB_DEV
     if agent_name == "WEB_DEV":
         agent = prompts.WEB_DEV
     if agent_name == "AI_SYSTEM_PROMPT":