Update app.py

app.py

@@ -1,9 +1,9 @@
 from transformers import pipeline
 import gradio as gr
 def story(StoryLength,StoryPrompt):
-    model= pipeline("text-generation", model="e-tony/gpt2-rnm")
+    model= pipeline("text-generation", model="EleutherAI/gpt-j-6B")
     summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-    return summarizer(model(StoryPrompt, max_length=StoryLength, num_return_sequences=1)[0]["generated_text"], max_length=130, min_length=30, do_sample=False)[0]["summary_text"]
+    return summarizer(model(StoryPrompt, max_length=StoryLength, num_return_sequences=1)[2]["generated_text"], max_length=130, min_length=30, do_sample=False)[0]["summary_text"]
     
     
 interface = gr.Interface(fn=story,