shivanshu292001 commited on
Commit
2dc124c
1 Parent(s): 3e9899e

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='gpt2')
8
  gen_blog=generator(txt,max_length=2000)
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='shivanshu292001/GeneratorModel')
8
  gen_blog=generator(txt,max_length=2000)
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))