Update app.py

app.py

@@ -18,9 +18,9 @@ def format_prompt(message, history):
 def generate(
         prompt, history, system_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, agent_name=None,
 ):
-    if agent_name = "WEB_DEV":
+    if agent_name == "WEB_DEV":
         agent = prompt.WEB_DEV
-    if agent_name = "AI_SYSTEM_PROMPT":
+    if agent_name == "AI_SYSTEM_PROMPT":
         agent = prompt.AI_SYSTEM_PROMPT
     system_prompt=agent
     temperature = float(temperature)