Spaces:
Runtime error
Runtime error
File size: 682 Bytes
406c499 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
tokenizer = GPT2Tokenizer.from_pretrained("gpt2-large")
model = GPT2LMHeadModel.from_pretrained("gpt2-large", pad_token_id=tokenizer.eos_token_id)
from transformers import pipeline, set_seed
generator = pipeline('text-generation', model='gpt2')
def generate_text(prompt):
text1 = generator(prompt, max_length=3000, num_return_sequences=1)
return text1[0].get('generated_text')
import gradio as gr
gr.Interface(
title = 'Text Generation using GPT 2',
fn=generate_text,
inputs=gr.Textbox(placeholder="Type Here..."),
outputs=[
"text"
],
theme = 'darkhuggingface').launch() |