Spaces:
Runtime error
Runtime error
File size: 816 Bytes
9b0697a a5bfd12 8d53e09 04d107a 9b0697a 01fdd70 a75de7e 9b0697a 73c11c9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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)
|