Update README.md
Browse files
README.md
CHANGED
@@ -16,7 +16,7 @@ tokenizer = AutoTokenizer.from_pretrained("Kwaai/GPT2_NonToxic")
|
|
16 |
model = AutoModelForCausalLM.from_pretrained("Kwaai/GPT2_NonToxic")
|
17 |
|
18 |
# Example usage
|
19 |
-
input_text = "
|
20 |
inputs = tokenizer(input_text, return_tensors='pt')
|
21 |
outputs = model.generate(**inputs)
|
22 |
|
|
|
16 |
model = AutoModelForCausalLM.from_pretrained("Kwaai/GPT2_NonToxic")
|
17 |
|
18 |
# Example usage
|
19 |
+
input_text = "you are toxic!"
|
20 |
inputs = tokenizer(input_text, return_tensors='pt')
|
21 |
outputs = model.generate(**inputs)
|
22 |
|