import gradio as gr
import json
from transformers import AutoTokenizer, AutoModelWithLMHead
from transformers import pipeline
tokenizer = AutoTokenizer.from_pretrained("gpt2")
model = AutoModelWithLMHead.from_pretrained("gpt2-large")
generator = pipeline(
  task="text-generation",
  model=model,
  tokenizer=tokenizer,
  framework="tf"
)
model = AutoModelWithLMHead.from_pretrained("gpt2")
def green(prompt):
    return generator(prompt, max_length=1000)
    
iface = gr.Interface(fn=green, inputs=gr.inputs.Textbox(lines=2, placeholder="Put prompt here"), outputs=gr.outputs.Textbox(type="str", label="generated text"))
iface.launch()