Zerx966 commited on
Commit
3a708d9
·
verified ·
1 Parent(s): b435e16

Delete Gpt

Browse files
Files changed (1) hide show
  1. Gpt +0 -14
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)