File size: 301 Bytes
2fedd68
7599980
2fedd68
202b83f
 
2fedd68
 
 
202b83f
 
 
 
01876a6
2fedd68
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import gradio as gr
from transformers import pipeline

pipeline = pipeline(task="text-generation", model="gpt2")

def greet(name):
    return "Hello " + name + "!!"

def gen_text(text):
    return pipeline(text)[0]
    
iface = gr.Interface(fn=gen_text, inputs="text", outputs="text")
iface.launch()