Update app.py

app.py

@@ -66,17 +66,19 @@ def send_chat_history_to_api(history):
         if response.status_code != 200:
             print(f"Failed to add data. Status code: {response.status_code}")
 
-sys_prompt = "FortiOS firewall configuration in CLI.\n"
+sys_prompt = "You are a knowledgeable AI assistant in FortiOS CLI.\
+        Your role is to assist users by providing accurate information and answering questions about the config in FortiOS CLI.\
+        Your answer should be clear and relevant.\n"
 
 if __name__ == "__main__":
     args = parse_args()
-    tokenizer = AutoTokenizer.from_pretrained("lliu01/fortios")
+    tokenizer = AutoTokenizer.from_pretrained("lliu01/fortios_cli")
     if not tokenizer.pad_token:
         tokenizer.pad_token = tokenizer.eos_token
         print(f"The tokenizer.pad_token set as a {tokenizer.eos_token}")
 
     model = AutoModelForCausalLM.from_pretrained(
-        "lliu01/fortios",
+        "lliu01/fortios_cli",
         torch_dtype=torch.bfloat16,
         low_cpu_mem_usage=True
     )