ganeshkamath89 commited on
Commit
e91e0ec
·
verified ·
1 Parent(s): c805802

Adding transformers library to import pipeline

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import gradio as gr
 
2
 
3
  def generate_story(story):
4
  story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")
 
1
  import gradio as gr
2
+ from transformers import pipeline
3
 
4
  def generate_story(story):
5
  story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")