shivanshu292001 commited on
Commit
3f6e731
1 Parent(s): d88df95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
 
5
 
6
  def blog_gen(txt):
7
- generator=pipeline(task='text-generation',model='GeneratorModel')
8
  gen_blog=generator(txt,max_length=300,num_return_sequences=2)
9
  remove_gen_text = list(map(lambda x: x["generated_text"], gen_blog))
10
  clean_text = list(map(lambda x: x.replace("\n\n", " "), remove_gen_text))
 
4
 
5
 
6
  def blog_gen(txt):
7
+ generator=pipeline(task='text-generation',model='gpt2')
8
  gen_blog=generator(txt,max_length=300,num_return_sequences=2)
9
  remove_gen_text = list(map(lambda x: x["generated_text"], gen_blog))
10
  clean_text = list(map(lambda x: x.replace("\n\n", " "), remove_gen_text))