Update app.py

app.py

@@ -14,9 +14,12 @@ def format_prompt(message, history):
     prompt += f" {bot_response}</s> "
   prompt += f"[INST] {message} [/INST]"
   return prompt
-
+agents =[
+    "WEB_DEV",
+    "AI_SYSTEM_PROMPT",
+]
 def generate(
-        prompt, history, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, agent_name=None,
+        prompt, history, sys_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, agent_name=agents[0],
 ):
     if agent_name == "WEB_DEV":
         agent = prompt.WEB_DEV
@@ -45,14 +48,11 @@ def generate(
         output += response.token.text
         yield output
     return output
-agents =[
-    "WEB_DEV",
-    "AI_SYSTEM_PROMPT",
-]
+
 
 additional_inputs=[
     gr.Textbox(
-        label="Prompt",
+        label="System Prompt",
         max_lines=1,
         interactive=True,
     ),
@@ -96,7 +96,7 @@ additional_inputs=[
     gr.Dropdown(
         label="Agents",
         choices=[s for s in agents],
-        value=agent[0],
+        value=agents[0],
         interactive=True,
         ),