File size: 183 Bytes
7b1f2d9
18d3078
36a627d
18d3078
7b1f2d9
18d3078
1
2
3
4
5
6
from transformers import pipeline
import gradio as gr
import torch

pipeline = pipeline("text-generation", model="openai-community/gpt2")
gr.Interface.from_pipeline(pipeline).launch()