sivang commited on
Commit
110e5f7
·
verified ·
1 Parent(s): ed4364c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +31 -1
README.md CHANGED
@@ -1,4 +1,34 @@
1
  ---
2
  license: gpl-3.0
3
  library_name: transformers
4
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: gpl-3.0
3
  library_name: transformers
4
+ language: en
5
+ tags:
6
+ - language-model
7
+ - text-generation
8
+ - security
9
+ - shell
10
+ ---
11
+
12
+ # SandboxLM
13
+
14
+ **SandboxLM** is a language model fine-tuned on a carefully curated synthetic dataset using the GPT-2 architecture. This model was specifically created to act as a special advisory to AI agents using shell commands, helping them operate securely by identifying potentially harmful shell commands. SandboxLM aims to assist in improving the safety and security of AI-driven shell operations.
15
+
16
+ ## Model Description
17
+
18
+ SandboxLM is built on the GPT-2 architecture, a Transformer-based language model. The model has been fine-tuned on a dataset designed to help identify and classify shell commands as either safe or potentially dangerous. This makes it suitable for security advisory tasks, particularly in environments where AI agents are used to execute shell commands.
19
+
20
+ ### Usage
21
+
22
+ To use this model, install the `transformers` library and load the model and tokenizer as follows:
23
+
24
+ ```python
25
+ from transformers import GPT2Tokenizer, GPT2LMHeadModel
26
+
27
+ tokenizer = GPT2Tokenizer.from_pretrained("your-username/sandboxlm")
28
+ model = GPT2LMHeadModel.from_pretrained("your-username/sandboxlm")
29
+
30
+ input_text = "rm -rf /"
31
+ inputs = tokenizer(input_text, return_tensors="pt")
32
+ outputs = model.generate(**inputs)
33
+
34
+ print(tokenizer.decode(outputs[0], skip_special_tokens=True))