ibrahimgiki commited on
Commit
6df2d37
·
verified ·
1 Parent(s): 156860e

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -0
app.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Load the GPT-2 large model and tokenizer
2
+ model_name = "gpt2-large"
3
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
4
+ model = AutoModelForCausalLM.from_pretrained(model_name)
5
+
6
+ # Function to generate a blog post based on a topic title
7
+ def generate_blog(topic_title, max_length=300):
8
+ # Step 1: Encode the input
9
+ input_ids = tokenizer.encode(topic_title, return_tensors='pt')
10
+
11
+ # Step 2: Generate model output
12
+ output_ids = model.generate(input_ids, max_length=max_length, num_return_sequences=1, pad_token_id=tokenizer.eos_token_id)
13
+
14
+ # Step 3: Decode the output
15
+ blog_post = tokenizer.decode(output_ids[0], skip_special_tokens=True)
16
+
17
+ return blog_post
18
+
19
+ # Example usage
20
+ topic_title = input("Enter a topic title for the blog post: ")
21
+ blog_post = generate_blog(topic_title)
22
+ print("\nGenerated Blog Post:\n")
23
+ print(blog_post)