ashirhashmi commited on
Commit
d8fb9a0
·
verified ·
1 Parent(s): 2e93adf

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +32 -0
  2. requirments.txt +1 -0
app.py ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # -*- coding: utf-8 -*-
2
+ """blogGenForHuggingFace.ipynb
3
+
4
+ Automatically generated by Colab.
5
+
6
+ Original file is located at
7
+ https://colab.research.google.com/drive/1_WrPpqQ6ep13S0Jiy-9bMyWea_EPEw4_
8
+ """
9
+
10
+ from transformers import pipeline, GPT2LMHeadModel, GPT2Tokenizer
11
+
12
+ # Load pre-trained GPT-2 model and tokenizer
13
+ model_name = "gpt2"
14
+ model = GPT2LMHeadModel.from_pretrained(model_name)
15
+ tokenizer = GPT2Tokenizer.from_pretrained(model_name)
16
+
17
+ # Define function to generate blog post
18
+ def generate_blogpost(topic):
19
+ input_text = f"Blog post about {topic}:"
20
+ input_ids = tokenizer.encode(input_text, return_tensors="pt")
21
+
22
+ # Generate text
23
+ output = model.generate(input_ids, max_length=500, num_return_sequences=1, no_repeat_ngram_size=2)
24
+
25
+ # Decode and return text
26
+ generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
27
+ return generated_text
28
+
29
+ # Example usage
30
+ topic = "natural language processing"
31
+ blogpost = generate_blogpost(topic)
32
+ print(blogpost)
requirments.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ transformers