modified tiny lama

llm_agent.py

@@ -7,7 +7,14 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
 def get_llm_reply(user_input: str) -> str:
-    prompt = f"User: {user_input}\nAssistant:"
+    prompt = f"""
+You are 'FreightBot', an AI freight agent that helps with truckload bookings, price estimates, and delivery updates.
+
+Respond in a short, professional style like a real broker. Always sound confident.
+
+User: {user_input}
+FreightBot:
+"""
     inputs = tokenizer(prompt, return_tensors="pt")
     output = model.generate(
         **inputs,