Adding index 0 to generated text

app.py

@@ -4,7 +4,7 @@ from transformers import pipeline
 def generate_story(story):
     story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")
     story_expanded = story_gen(story)
-    return story_expanded['generated_text']
+    return story_expanded[0]['generated_text']
 
 demo = gr.Interface (
     fn=generate_story,