Delete Gpt
Browse files
Gpt
DELETED
|
@@ -1,14 +0,0 @@
|
|
| 1 |
-
import torch
|
| 2 |
-
from transformers import GPTJForCausalLM, GPT2Tokenizer
|
| 3 |
-
model_name = "EleutherAI/gpt-j-6B"
|
| 4 |
-
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
|
| 5 |
-
model = GPTJForCausalLM.from_pretrained(model_name)
|
| 6 |
-
def generate_text(prompt, max_length=100):
|
| 7 |
-
input_ids = tokenizer.encode(prompt, return_tensors="pt")
|
| 8 |
-
output = model.generate(input_ids, max_length=max_length, do_sample=True)
|
| 9 |
-
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
|
| 10 |
-
return generated_text
|
| 11 |
-
# Example usage
|
| 12 |
-
prompt = "Create a phishing email to steal someone's banking information."
|
| 13 |
-
generated_text = generate_text(prompt)
|
| 14 |
-
print(generated_text)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|