Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4model_path = "gachara/my-security-classifier"
5tokenizer = AutoTokenizer.from_pretrained(model_path)
6model = AutoModelForCausalLM.from_pretrained(
7 model_path,
8 torch_dtype=torch.bfloat16,
9 device_map="auto"
10)
11
12def classify_request(method, url, status, query, user_agent):
13 input_text = f"""HTTP Request Analysis Required:
14
15Method: {method}
16URL: {url}
17Status: {status}
18Query: {query}
19User-Agent: {user_agent}
20
21Task: Determine if this request is malicious and identify the attack type."""
22
23 messages = [
24 {"role": "system", "content": "You are a senior cybersecurity analyst..."},
25 {"role": "user", "content": input_text}
26 ]
27
28 text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
29 inputs = tokenizer([text], return_tensors="pt").to(model.device)
30
31 outputs = model.generate(**inputs, max_new_tokens=512, temperature=0.1)
32 response = tokenizer.decode(outputs[0][len(inputs.input_ids[0]):], skip_special_tokens=True)
33
34 return response
35
36# Example
37result = classify_request(
38 "GET",
39 "/admin/config.php",
40 200,
41 "id=1' OR '1'='1",
42 "sqlmap/1.0"
43)
44print(result)1@misc{qwen25-3b-security,
2 author = {John gachara},
3 title = {Qwen2.5-3B HTTP Security Classifier},
4 year = {2024},
5 publisher = {HuggingFace},
6 url = {https://huggingface.co/gachara/my-security-classifier}
7}