Spaces:
Runtime error
Runtime error
| from transformers import pipeline | |
| import gradio as gr | |
| def story(StoryLength,StoryPrompt): | |
| model= pipeline("text-generation", model="e-tony/gpt2-rnm") | |
| summarizer = pipeline("summarization", model="facebook/bart-large-cnn") | |
| return model(StoryPrompt, max_length=200, num_return_sequences=3)[2]["generated_text"],summarizer(model(StoryPrompt, max_length=200, num_return_sequences=3)[2]["generated_text"], max_length=StoryLength, min_length=30, do_sample=False)[0]["summary_text"] | |
| interface = gr.Interface(fn=story, | |
| inputs=["number","text"], | |
| outputs=[gr.inputs.Textbox(label='GPT2 Output'),gr.inputs.Textbox(label='Output summary')], | |
| title='Bilal\'s Story Generator') | |
| interface.launch(inline=False) | |