Spaces:
Runtime error
Runtime error
File size: 574 Bytes
d169e0f 9502b13 64ca7f4 9502b13 318a832 0c176fd 318a832 9502b13 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from transformers import AutoTokenizer, AutoModelForCausalLM,pipeline
import gradio as gr
tokenizer = AutoTokenizer.from_pretrained("distilgpt2")
model = AutoModelForCausalLM.from_pretrained("distilgpt2")
gen_pipeline = pipeline(task = "text-generation", model=model, tokenizer=tokenizer)
get_generated_text = lambda x: gen_pipeline(x,max_length = 200)[0]['generated_text']
demo = gr.Interface(
fn = get_generated_text ,
inputs = gr.Textbox(label = 'Enter A series of Text and Generate more',lines = 2, max_lines = 5),
outputs = 'text',
)
demo.launch() |