File size: 351 Bytes
f7ffb15 71a21b8 f7ffb15 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from transformers import pipeline
import gradio as gr
import os
token = os.environ["HF_TOKEN"]
pipe = pipeline("text-generation", model="openai-community/gpt2")
def generate(text):
return pipe(text)[0]["generated_text"]
demo = gr.Interface(fn=generate,
inputs=["text"],
outputs=["text"])
demo.launch() |